hrbust 哈理工oj 2181 无聊的小明【数位DP】

最新推荐文章于 2021-03-14 22:57:01 发布

mengxiang000000

最新推荐文章于 2021-03-14 22:57:01 发布

阅读量681

点赞数

分类专栏： dp 文章标签： hrbust 哈理工oj 2181

本文链接：https://blog.csdn.net/mengxiang000000/article/details/50908859

版权

dp 专栏收录该内容

351 篇文章 2 订阅

订阅专栏

无聊的小明

Time Limit: 3000 MS

Memory Limit: 32768 K

Total Submit: 49(24 users)

Total Accepted: 27(22 users)

Rating:

Special Judge: No

Description

小明想用两个字母a和b创造一个长度为n的单词，但是他又不希望连续的a超过p个，同时也不希望连续的b超过q个。那么小明有能创造出多少个不用的单词呢？

Input

每组数据包括一行，三个整数n，p，q分别对应题意。

其中max(a, b) <= n <= 50000，1 <= a, b <= 300。

Output

输出不同的单词的个数。个数要对1000000007取模。

Sample Input

3 2 1

Sample Output

Source

2014.11.29新生赛-热身赛

我们设DP【i】【j】表示长度为i的字符串最后一个字符为j，其中因为只有a，b两个字母，所以我们令j=2；

所以开到dp【50005】【2】即可。

我们要保证连续a字母不超过p个，那么dp【i】【0】=sum（dp【i】【1】（i-p<=i<=n-1））。

我们要保证连续b字母不超过q个，那么dp【i】【1】=sum（dp【i】【0】（i-q<=i<=n-1））。

那么长度为i并且满足p，q的情况下，dp【i】【0】+dp【i】【1】==长度为i的时候满足p，q的组合总数。

AC代码：

#include<stdio.h>
#include<string.h>
using namespace std;
#define mod 1000000007
#define ll long long int
ll dp[50005][2];
int main()
{
    int n,p,q;
    while(~scanf("%d%d%d",&n,&p,&q))
    {
        memset(dp,0,sizeof(dp));
        dp[0][1]=1;
        dp[0][0]=1;
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=i&&j<=p;j++)
            {
                dp[i][1]+=dp[i-j][0];
                dp[i][1]%=mod;
            }
            for(int j=1;j<=i&&j<=q;j++)
            {
                dp[i][0]+=dp[i-j][1];
                dp[i][0]%=mod;
            }
        }
        printf("%lld\n",(dp[n][0]+dp[n][1])%mod);
    }
}