51Nod 1043 幸运号码 数位DP

1043 幸运号码

 

基准时间限制：1 秒 空间限制：131072 KB 分值: 20 难度：3级算法题

 收藏

 关注

1个长度为2N的数，如果左边N个数的和 = 右边N个数的和，那么就是一个幸运号码。

例如：99、1230、123312是幸运号码。

给出一个N，求长度为2N的幸运号码的数量。由于数量很大，输出数量 Mod 10^9 + 7的结果即可。

Input

输入N(1<= N <= 1000)

Output

输出幸运号码的数量 Mod 10^9 + 7

Input示例

1

Output示例

9

李陶冶 (题目提供者)

 

 

关于计数的题目，数位DP

我们用 dp[i][j] 代表i位数的和为j的方案数

一位数 xxxt 可以看做是在一个三位数后面添加一个数t 

明显 dp[i][j] + = dp[i-1][j-t];

转移 就是dp[i][j]=Σdp[i-1][j-t] (j-t>=0) 

统计ans要注意 

每次我们都是在一个数后面加一个数t(t>=0&&t<=9) 

所以我们可能出现 左边数的第一位我们放了0的情况 

左边的一个数的第一位不可能为0 所以要减去第一位为0的情况

 

#include <cctype>
#include <cstdio>

typedef long long LL;
const int mod=1e9+7;
const int MAXN=1010;

int n,m;

LL ans;

LL dp[2][MAXN*10];

int hh() {
    scanf("%d",&n);
    dp[0][0]=1;
    for(int i=1;i<=n;++i)
      for(int j=0;j<=n*9;++j) {
          dp[i&1][j]=0;
            for(int t=0;t<=9;++t) {
              if(j-t<0) continue;
              dp[i&1][j]=(dp[i&1][j]+dp[(i-1)&1][j-t])%mod;
          }
      }
    for(int i=0;i<=n*9;++i) 
      ans=(ans+dp[n&1][i]*(dp[n&1][i]-dp[(n-1)&1][i]))%mod;
    printf("%lld\n",ans);
    return 0;
}

int sb=hh();
int main(int argc,char**argv) {;}