51nod-1043 幸运号码

[题目链接]

思路：刚开始一看数位dp，好久没写了，递归模板拿出来摸一摸，然后掉进坑里了无限TLE，根本不需要啊！再然后，看了一下题解和自己想的差不多，一直写一直写，无限找bug，取模mod=1e9+10，麻蛋，加毛线的10啊~~

用dp[i][j]表示i个数的和为j的总数，这里面是包括0开头的情形，有dp[i][j]=dp[i-1] [j-k] (k从0到9)。
很好想，i个数组成总和为j的数量就来自于i-1个数 里面能 在最前面加0到9的数字使得加完之后和为j。
这里面包含了0开头的，把0去掉的方法就是dp[i][j]-dp[i-1][j]。
dp[i-1][j]就代表了在i个数中，开头为0的个数，减去就是i个数中开头不为0的个数。
原因很明显，i个数和为j与i-1个数和为j，就差了一个位置为0。而这一个位置因为一开始咱们的想法就是在最前面加的数字，所以这个位置就差在了最前面的位置上。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int mod=1e9+7;

int n;
ll dp[2][10000];

int main()
{
    scanf("%d",&n);
    memset(dp,0,sizeof(dp));
    dp[0][0]=1; //0个数 凑成0的情况是1
    for(int i=0;i<=9;i++)dp[1][i]=1;
    for(int i=2;i<=n;i++){
        for(int j=0;j<=i*9;j++){
            dp[i&1][j]=0;
            for(int k=0;k<=9;k++)
            if(j>=k)dp[i&1][j]=(dp[i&1][j]+dp[(i-1)&1][j-k])%mod;
        }
    }
    ll sum=0;
    for(int i=0;i<=9*n;i++)
        sum=(sum+dp[n&1][i]*(dp[n&1][i]-dp[(n-1)&1][i]))%mod;
    printf("%lld\n",sum);
    return 0;
}