51nod数字1的数量

这道题瞎jbyy了很久 方法可能很奇怪...

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define LL long long
using namespace std;
int n,ans;
int f[35][10],w[11];
void init(){
w[1]=1;
for(int i=2;i<=10;i++) w[i]=w[i-1]*10;
f[1][1]=1;
for(int i=2;i<=10;i++)
for(int j=0;j<=9;j++){
for(int k=0;k<=9;k++)f[i][j]+=f[i-1][k];
if(j==1) f[i][j]+=w[i];
}
//for(int i=1;i<=10;i++,printf("\n")) for(int j=0;j<=9;j++) printf("%d ",f[i][j]);
}
int main()
{
init();
int sum=10,ans=0,v,k,tot=2;
scanf("%d",&n); k=n;
while(w[sum]>n) sum--;
int now=n/w[sum];
for(int i=0;i<now;i++) ans+=f[sum][i];
n=n%w[sum];
for(int i=sum-1;i;i--){
v=n/w[i];
for(int j=0;j<v;j++) ans+=f[i][j];
n=n%w[i];
}
v=k;
if(k%10==1) ans++;
v=v/10;
while(tot<=sum){
if(v%10==1) ans=ans+k%w[tot]+1;
tot++; v=v/10;
}
printf("%d\n",ans);
return 0;
}