51nod 1043 幸运号码【dp+滚动数组】

最新推荐文章于 2017-11-03 15:24:36 发布

原创最新推荐文章于 2017-11-03 15:24:36 发布 · 668 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#51nod 1043

思维同时被 2 个专栏收录

692 篇文章

订阅专栏

351 篇文章

订阅专栏

本文介绍了一种通过动态规划算法解决寻找特定长度的幸运号码数量的问题。幸运号码是指一个长度为2N的数，其中左边N个数的和等于右边N个数的和。文章详细阐述了动态规划的状态定义、状态转移方程，并提供了去除前导0情况的解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1043 幸运号码

基准时间限制：1 秒空间限制：131072 KB 分值: 20 难度：3级算法题

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

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

给出一个N，求长度为2N的幸运号码的数量。由于数量很大，输出数量 Mod 10^9 + 7的结果即可。

Input

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

Output

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

Input示例

Output示例

思路：

1、计数问题，考虑dp：

①我们肯定是根据数位进行dp，那么设定dp【i】【j】表示现在一个数，一共长度为i，和为j的方案数。

②那么不难推出其状态转移方程：
dp【i】【j】=dp【i】【j】+dp【i-1】【j-k】，其中k表示当前第i位最后一个数字为k。

③对于状态转移过程中，注意去掉前导0的情况发生。

2、那么我们答案就是Σdp【n】【i】*dp【n】【i】？

很显然不是，因为我们要得到的是长度为2n的一个数，而非两个不含前导0的部分连接在一起，我们只要保证前边n个数所组成的方案中，不包含前导0即可，后半部分是可以有前导0存在的，所以我们再设定一个dp2【i】【j】，状态转移方程一样，只是这个部分我们不对前导0进行处理。

那么答案就是：Σdp【n】【i】*dp2【n】【i】（1<=i<=n*9）；

3、因为我们设定的数组偏大，而又考虑到第i位只和第i-1位方案数有关，那么我们考虑用滚动数组进行空间优化。

另外，因为思路比较挫，暴力姿势如果不太好的话会TLE掉、注意暴力姿势。

4、其实这个题打表也是个很好的选择，暴力Dp反正也不会多久，跑个表直接打印粗来。。。。（继续膜效率榜第一的方法.................）

Ac代码：

#include<stdio.h>
#include<string.h>
using namespace std;
#define mod 1000000007
#define ll __int64
ll dp[2][9050];
ll dp2[2][9050];
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        memset(dp,0,sizeof(dp));
        memset(dp2,0,sizeof(dp2));
        dp[0][0]=1;
        dp2[0][0]=1;
        for(int i=1;i<=n;i++)
        {
            for(int j=0;j<=i*9;j++)
            {
                for(int k=0;k<=9;k++)
                {
                    int q=j-k;
                    if(q>=0)
                    dp2[1][j]=(dp2[1][j]+dp2[0][q])%mod;
                    if(i==1&&k==0)continue;
                    if(q>=0)
                    dp[1][j]=(dp[1][j]+dp[0][q])%mod;
                }
            }
            for(int j=0;j<=i*9;j++)
            {
                dp[0][j]=dp[1][j],dp2[0][j]=dp2[1][j];
                dp[1][j]=0,dp2[1][j]=0;
            }
        }
        ll output=0;
        for(int i=1;i<=n*9;i++)
        {
            output=(output+(dp[0][i]*dp2[0][i])%mod)%mod;
        }
        printf("%I64d\n",(output+mod)%mod);
    }
}