51nod 1009 数字1的数量(数位dp)

基础的数位dp，我不大会。
数位dp的ppt：https://wenku.baidu.com/view/9de41d51168884868662d623.html
建议先看一下ppt，我是按照ppt里面差不多的方式写的，也参考了别人写的。dp[i][j]表示长度为i位以j开头的数字到1（闭区间）之间包含多少1

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

int dp[15][15];
int digit[15];
int dlen;

void init()
{
    for(int i = 1; i <= 10; ++i)
    {
        for(int j = 0; j < 10; ++j)
        {
            if(j == 1)
                dp[i][j+1] = dp[i-1][9+1]*2+pow(10,i-1);
            else
                dp[i][j+1] = dp[i-1][9+1]+dp[i][j-1+1];
        }
    }
}

void toDight(int n)
{
    dlen = 0;
    while(n)
    {
        digit[++dlen] = n%10;
        n /= 10;
    }
}

int query(int n)
{
    int res = 0;
    for(int i = dlen; i >= 1; --i)
    {
        if(digit[i] == 0) continue;
        if(digit[i] == 1)
            res += dp[i-1][9+1]+1+n%((int)pow(10,i-1));
        else
            res += dp[i][digit[i]];
    }
    return res;
}

int main()
{
    init();
    int n;
    scanf("%d",&n);

    toDight(n);
    printf("%d\n",query(n));
    return 0;
}

下边这写法是参考的：http://blog.csdn.net/f_zyj/article/details/52072686
参考代码里给的注释讲的挺好，我是单步了几遍才看懂的。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

//dp[i]表示i为数字到1之间一共包含的1的个数
int dp[15];

void init()
{
    for(int i = 1; i <= 9; ++i)
        dp[i] = dp[i-1]*10+pow(10,i-1);
}

int main()
{
    init();
    int n;
    scanf("%d",&n);
    int num = n;
    int res = 0;
    int cur;
    int len = 0;
    int ten = 1;
    int tail = 0;
    while(num)
    {
        cur = num%10;
        num /= 10;
        len++;
        if(cur > 1)
        {
            res = res + dp[len-1]*cur + ten;
        }
        else if(cur == 1)
        {
            res = res + dp[len-1] + 1 + tail;
        }
        tail = tail + cur*ten;
        ten *= 10;
    }
    printf("%d\n",res);
    return 0;
}