1043 幸运号码 （简单数位dp）

1043 幸运号码

 

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

个幸运号码。

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

给出一个N，求长度为2N的幸运号码的数量。由于数量很大，输出数量

Mod 10^9 + 7的结果即可。

输入

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

输出

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

输入样例

1

输出样例

9

思路：

      这题一开始考虑前后两个N位数重复会造成答案冗余，其实

是我自己考虑错了，计算出N位数构造和为sum有多少种方案，

然后 不带前导零的方案数 * 带前导零的方案数即可。

     dp数组第二维范围开小了，WA，自己一直考虑，这题答案

要不要处理负数模的情况，不处理也能过，但应该处理吧？还是

不会出现负数情况？

代码实现：

#include <iostream>
#include <string.h>
#include <math.h>
#include <ctime>
#include <bits/stl_algo.h>
#include <stdio.h>
#include <algorithm>

#define LL long long
#define INF 0x3f3f3f3f
#define ull unsigned long long

using namespace std;

const int N = 2e5 + 100;
const int mod = 1e9 + 7;

long long dp[1010][9200];

int main ()
{

#ifdef MYHOME_Wjvje
    freopen("input.txt", "r", stdin);
#endif
    int n;
    long long ans;
    while (cin >> n)
    {
        ans = 0;
        memset(dp, 0, sizeof(dp));
        for (int i = 0; i <= 9; i ++) dp[1][i] = 1;
        for (int i = 2; i <= n; i ++)
            for (int j = 0; j <= 9 * i; j ++)
                for (int k = 0; k <= 9 && k <= j; k ++)
                    dp[i][j] = (dp[i][j] + dp[i-1][j-k]) % mod;

        for (int j = 1; j <= 9 * n; j ++)
        	ans = (ans + dp[n][j] * (dp[n][j] - dp[n-1][j]) 
% mod + mod) % mod;
        cout << ans << endl;
    }
    return 0;
}

 THE END;