【51NOD1009数字1的数量】 数位DP

51NOD1009数字1的数量
题意就是求1-n之内出现过多少个1，比如11出现两次，112出现两次。
直接分为两种数位dp，
当当前位为1时，后面进行无限制数位dp求方案数，
当前位不为1时，直接数位dp下一位，
这样就算出了每一位为1对前一位的贡献。前一位不断加上后面可能的贡献数就好了。
代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
using namespace std;
typedef long long ll;
int a[20];
ll dp[20];
ll dp2[20];
ll dfs2(int pos,bool limit)
{
    if(pos==-1) return 1;
    if(!limit && dp2[pos]!=-1) return dp2[pos];
    int up=limit?a[pos]:9;
    ll tmp=0;
    for(int i=0;i<=up;i++)
    {
        tmp+=dfs2(pos-1,limit && i==a[pos]);
    }
    if(!limit) dp2[pos]=tmp;
    return tmp;
}
ll dfs(int pos,bool limit)
{
    if(pos==-1) return 0;
    if(!limit && dp[pos]!=-1) return dp[pos];
    int up=limit?a[pos]:9;
    ll tmp=0;
    for(int i=0;i<=up;i++)
    {
        if(i==1) tmp+=dfs2(pos-1,limit && i==a[pos]);
        tmp+=dfs(pos-1,limit && i==a[pos]);
    }
    if(!limit) dp[pos]=tmp;
    return tmp;
}
ll solve(int x)
{
    int pos=0;
    while(x)
    {
        a[pos++]=x%10;
        x/=10;
    }
    return dfs(pos-1,true);
}
int main()
{
    int le,ri;
    memset(dp,-1,sizeof dp);
    memset(dp2,-1,sizeof dp2);
    int n;
    scanf("%d",&n);
    printf("%lld\n",solve(n));
    return 0;
}