其他分享
首页 > 其他分享> > 1043 幸运号码 (简单数位dp)

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;

标签:1043,号码,int,long,幸运,include,dp,数位
来源: https://blog.csdn.net/qq_41661919/article/details/99651400