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
思路:
i位数的最大和为i*9。用dp[i][j]表示i位数的和为j。则dp[i][j] = sum(dp[i-1][j-k]), k = 0--9。因为dp[i][j]只和dp[i-1][j-k]有关,所以可以用滚动数组来表示dp[i][j]和dp[i-1][j-k]。计算乘积的时候注意n位数的开始不能为零,所以左边高位要把为零的情况排除,方法是:dp[n & 1][i] - dp[(n - 1) & 1][i]。
#include <iostream>
#include <cstring>
using namespace std;
typedef long long int ll;
#define MAXNUM 1005
const int MOD = 1e9 + 7;
ll dp[2][9*MAXNUM];
int main(int argc, const char * argv[])
{
int n;
cin >> n;
memset(dp, 0, sizeof(dp));
dp[0][0] = 1;
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++)
{
ll temp = 0;
for (int k = 0; k <= 9; k++)
{
if (j-k >= 0)
{
temp = (temp + dp[(i - 1) & 1][j - k]) % MOD;
}
else
{
break;
}
}
dp[i & 1][j] = temp;
}
}
ll result = 0;
for (int i = 0; i <= 9*n; i++)
{
result = (result + dp[n & 1][i] *(dp[n & 1][i] - dp[(n - 1) & 1][i])) % MOD;
}
cout << result << endl;
return 0;
}