ural 2018. The Debut Album 滚动数组dp

2018. The Debut Album

3 2 1

4

//0.562	210 KB
#include<stdio.h>
#include<string.h>
#include<algorithm>
#define M 1000000007
#define ll long long
using namespace std;
ll dp[2][2][307];
int main()
{
    int n,a,b,now,pre;
    while(scanf("%d%d%d",&n,&a,&b)!=EOF)
    {
        memset(dp,0,sizeof(dp));
        dp[0][0][0]=1;dp[0][1][0]=1;
        for(int i=1;i<=n;i++)
        {
            now=i&1,pre=(i-1)&1;
            memset(dp[now],0,sizeof(dp[now]));
            for(int j=1;j<=a;j++)
            {
                dp[now][0][j]+=dp[pre][0][j-1];
                if(dp[now][0][j]>M)dp[now][0][j]%=M;
                dp[now][1][1]+=dp[pre][0][j];
                if(dp[now][1][1]>M)dp[now][1][1]%=M;
            }
            for(int j=1;j<=b;j++)
            {
                dp[now][1][j]+=dp[pre][1][j-1];
                if(dp[now][1][j]>M)dp[now][1][j]%=M;
                dp[now][0][1]+=dp[pre][1][j];
                if(dp[now][0][1]>M)dp[now][0][1]%=M;
            }
        }
        ll ans=0;
        for(int i=1;i<=a;i++)
        {
            ans+=dp[n&1][0][i];
            if(ans>M)ans%=M;
        }
        for(int i=1;i<=b;i++)
        {
            ans+=dp[n&1][1][i];
            if(ans>M)ans%=M;
        }
        printf("%lld\n",ans);
    }
    return 0;
}