51nod 1043 幸运号码（数位dp

1043 幸运号码

 

 

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[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 <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod = 1e9+;
int dp[][];// dp[i][j] 为i个数 和为j的情况

int main()
{
    int n;
    scanf("%d",&n);
    dp[][] = ;//这里明明是网上题解有错误 他们都写的是dp[0][1] = 1
    //明明他们是在凑样例的感觉  应该是0个数 凑成0的情况是1
    for(int i=;i<=;i++)
        dp[][i] = ;
    for(int i=;i<=n;i++)
    {
        for(int j=;j<=*i;j++)
        {
            int sum = ;
            for(int k=;k<=;k++)
            {
                if(j >=k)
                    sum = (sum + dp[i-][j-k])%mod;
                else
                    break;
            }
            dp[i][j] = sum;
        }
    }
    ll ans = ;
    for(int i=;i<=*n;i++)
        ans = (ans + (ll)dp[n][i] * (dp[n][i]-dp[n-][i]))%mod;//这里单独 (ll)dp[n][i] * (dp[n][i]-dp[n-1][i]) 会超精度
    printf("%lld\n",ans);
}

下面是用滚动数组内存优化过的 用的now 和pre   （强行装逼， 不过倒腾了半个小时  now 和 pre 的关系

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod = 1e9+;

int dp[][];

int main ()
{
    int n;
    scanf("%d",&n);
    int now = ,pre = ;
    dp[][] = ;
    swap(now,pre);
    for(int i=;i<=;i++)
        dp[now][i] = ;
    swap(now,pre);
    for(int i=;i<=n;i++)
    {
        for(int j=;j<=*n;j++)
        {
            ll sum =;
            for(int k=;k<=;k++)
            {
                if(j>=k)
                    sum = (sum+dp[pre][j-k])%mod;
                else
                    break;
            }
            dp[now][j] = sum;
        }
        swap(now,pre);
    }
    ll ans = ;
    for(int i=;i<=*n;i++)
    {
        ans = (ans + (ll)dp[pre][i]*(dp[pre][i] - dp[now][i]))%mod;
    }
    printf("%lld\n",ans);
}

别人的代码  （随便看看就好 主要是滚动优化 可以学一下（n&1）

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define maxn 10005
#define Mod  1000000007
ll dp[][maxn];
int main()
{
    ll n,m,i,j,k,sum,ans=;
    scanf("%lld",&n);
    dp[][]=;//这里是0
    for(i=;i<=;i++)
    dp[][i]=;
    for(i=;i<=n;i++)
    {
        for(j=;j<=n*;j++)
        {
           sum=;
           for(k=;k<=;k++)
           {
                if(j>=k)
                sum=(sum+dp[(i-)%][j-k])%Mod;
                else
                dp[i%][j]=;
           }
           dp[i%][j]=sum;
        }
    }
    for(i=;i<=*n;i++)
    ans=(ans+dp[n%][i]*(dp[n%][i]-dp[(n-)%][i])%Mod)%Mod;
    printf("%lld\n",ans);
}