Codeforces #594 (Div. 1) A. Ivan the Fool and the Probability Theory（DP）

最新推荐文章于 2022-07-13 02:01:45 发布

路西菲尔

最新推荐文章于 2022-07-13 02:01:45 发布

阅读量145

点赞数

分类专栏：题解

本文链接：https://blog.csdn.net/qq_43781431/article/details/105157947

版权

题解专栏收录该内容

89 篇文章 0 订阅

订阅专栏

原题链接：https://codeforces.com/contest/1239/problem/A
大佬用来热身的小题
而我却迟迟想不出来

当n和m都为1的时候显然只有两种情况
当n为1，m不为1的时候或m为1，n不为1的时候
相当于在一行中每个方块进行黑白染色，且不能有三个连着的相同的颜色
d[i][1]表示第 i 块染白色的方案数，d[i][2]表示第 i 块染黑色的方案数
显然d[1][1]=1 d[1][2]=1 d[2][1]=2 d[2][2]=2
前两块怎么染都无所谓的，再往后
d[i][1]=d[i-1][2]+d[i-2][2]
d[i][2]=d[i-1][1]+d[i-2][1]
第 i 块染白色的方案数相当于第 i-1 块染黑色的方案数加上第 i-2 块染黑的方案数（仔细思考一下）

当n和m都大于2时转化一下也就相当于一行中染色
注意有重复的情况

代码：

#include <bits/stdc++.h>
#define ll long long
using namespace std;

ll d[200005][3],f[200005][3];
ll ans=0,mod=1e9+7;

int main()
{
    ll n,m;
    cin>>n>>m;
    if(n==1&&m==1){cout<<"2"<<endl;return 0;}
    d[1][1]=1;d[1][2]=1;
    d[2][1]=2;d[2][2]=2;
    for(int i=3;i<=n;i++)
    {
        d[i][1]=(d[i-1][2]+d[i-2][2])%mod;
        d[i][2]=(d[i-1][1]+d[i-2][1])%mod;
    }
    f[1][1]=1;f[1][2]=1;
    f[2][1]=2;f[2][2]=2;
    for(int i=3;i<=m;i++)
    {
        f[i][1]=(f[i-1][2]+f[i-2][2])%mod;
        f[i][2]=(f[i-1][1]+f[i-2][1])%mod;
    }
    if(n==1)
    {
        cout<<(f[m][1]+f[m][2])%mod<<endl;
        return 0;
    }
    if(m==1)
    {
        cout<<(d[n][1]+d[n][2])%mod<<endl;
        return 0;
    }
    cout<<(d[n][1]+d[n][2]+f[m][1]+f[m][2]-2)%mod<<endl;
    return 0;
}