51 nod 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

因为是求和 数字比较小所以可以直接开一维表示形成当前值有多少种可能

#include <iostream>#include <bits/stdc++.h>using namespace std;typedef  long long LL;const LL mod = 1e9+7;LL dp[3][10000];int main(){    int n;    scanf("%d", &n);    memset(dp,0,sizeof(dp));    for(int i=0;i<10;i++) dp[1][i]=1;    if(n==1)    {        cout<<9<<endl;        return 0;    }    int o=1;    for(int i=2;i<=n;i++)    {        for(int k=0;k<=i*9;k++)        {            LL sum=0;            for(int j=0;j<=9&&j<=k;j++)  sum=(dp[(i-1)&1][k-j]+sum)%mod;            dp[i&1][k]=sum;        }    }    LL ans=0;    for(int i=0;i<=n*9;i++)        ans=(ans+dp[n&1][i]*(dp[n&1][i]-dp[(n-1)&1][i]+mod)%mod)%mod;    cout<<ans<<endl;    return 0;}