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