hrbust 2181

数位dp

动态规划 的一种用法

dp[i][j]

i代表第几位j代表的这个位置元素的名称

我们需要遍历除在p，q内不同的元素加起来到i位

无聊的小明

Time Limit: 3000 MS Memory Limit: 32768 K Total Submit: 63(34 users) Total Accepted: 37(32 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

4

Source

2014.11.29新生赛-热身赛

Submit

Statistic

Discuss

Sharedcodes

#include<stdio.h>
#include<string.h>
#define mod 1000000007
long long dp[50003][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][1]+dp[n][0])%mod);
    }
}