当前位置:编程学习 > C/C++ >>

幸运号码-dp

题目:
时间限制:1 秒 空间限制:65536 KB 分值: 20
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
题目分析:
N比较小,仔细分析一下题目,有点像动态规划。想了好久没整明白递推方程,没有完整的写出来,更解决不了这道题目。
在qq上问了一下,一个人很快就列出了动态方程太牛了,然后自己分析了一下,没有问题,结果代码实现,最终也是没有问题,
这道题目,不能直接求出答案,需要分两步:第一步,先求出n位数,和为s的情况种数。这个相信很多人能够列出状态方程,类似于背包吧,
dp[n][s]表示n位,和为s的号码个数,方程为dp[n][s] = 求和(dp[n-1][s-i],i=[0,9])什么意思呢?也就是当第n位为0-9时,和为s的种类和n-1位,和为s减去0-9的种类有关,是求和的关系。
那么得到了上面的结果后,如何求得N位的幸运号码呢?N位,和为s的种类数知道了,但是2N位要求前N位的首位不能为0,而后N位无所谓,那么首位不为0的个数怎么确定呢,这里有个很巧妙的办法,就是dp[n][s] - dp[n-1][s]这就是第n位为0的情况,那么有了这个就直接可以求出所有的幸运号码了就是dp[n][s]*(dp[n][s] - dp[n-1][s]),s=[0,9*N],求和就是结果。
有了这些分析,就能够写程序了,注意,这里第n位只和第n-1位有关,所以只需要开辟一个2*N*9的空间就行了,滚动数组,并且要int64,否则会溢出。
下面就直接是代码了,好好体会动态规划的分析过程,为何自己第一时间想不到。。。。很感谢这位好友,这里分享一下思路,以作总结啦。
代码:
 
#include <iostream>  
#include <cstdlib>  
#include <cstring>  
//#include <Windows.h>  
using namespace std;  
const int MAXN = 1010;  
const int MAX = 1000000007;  
  
__int64 dp[2][9*MAXN];  
  
int main()  
{  
    int N;  
    while (cin >> N)  
    {  
        //int time = GetTickCount();  
        memset(dp,0,sizeof(dp));  
        dp[1][0] = 1;  
        for (int i = 0; i <= 9; ++ i)  
        {  
            dp[0][i] = 1;  
        }  
        for (int i = 1; i < N; ++ i)  
        {  
            for (int s = 0; s <= 9 * (i+1); ++ s)  
            {  
                __int64 sum = 0;  
                for (int j = 0; j <= 9; ++ j)  
                {  
                    if (s >= j)  
                    {  
                        sum = (sum + dp[(i - 1)%2][s - j])%MAX;  
                    }  
                    else  
                        dp[i%2][s] = 0;  
                }  
                dp[i%2][s] = sum;  
            }  
        }  
        __int64 ans = 0;  
        for (int i = 0; i <= N * 9; ++ i)  
        {  
            ans = (ans + dp[(N+1)%2][i]*(dp[(N+1)%2][i] - dp[N%2][i]))%MAX;  
        }  
  
        cout << ans << endl;  
    }  
}  

 

 
 
补充:软件开发 , C++ ,
CopyRight © 2022 站长资源库 编程知识问答 zzzyk.com All Rights Reserved
部分文章来自网络,