【AT2567】RGB Sequence

中文題面:洛谷AT2567,在討論中有翻譯。

解法:動態規劃

fi,j,k 表示已填了前 i 位置的顏色,三種顏色的最後出現位置分別爲 i,j,k (注意無序),轉移爲:

{fi,j,kfi+1,i,jfi,j,kfi+1,i,kfi,j,kfi+1,j,k

最後注意用好 jk 的意義將不合法的狀態位置設爲 0

代碼:

#include<iostream>
#include<cstdio>
#include<vector>

#define int long long

using namespace std;

const int p=1000000007;
int f[302][301][301],N,M,l,r,x,ans;
vector<pair<int,int>> vec[301];

main(){
    scanf("%lld%lld",&N,&M);for(int i=1;i<=M;++i)scanf("%lld%lld%lld",&l,&r,&x),vec[r].push_back(make_pair(l,x));
    for(int i=f[1][0][0]=1;i<=N;++i){
        for(pair<int,int> pr:vec[i])for(int j=0;j<i;++j)for(int k=0;k<=(j>0?j-1:j);++k)if(pr.second==1){if(pr.first<=j)f[i][j][k]=0;}else if(pr.second==2){if(pr.first<=k||j<pr.first)f[i][j][k]=0;}else if(k<pr.first)f[i][j][k]=0;
        for(int j=0;j<i;++j)for(int k=0;k<=(j>0?j-1:j);++k)(f[i+1][j][k]+=f[i][j][k])%=p,(f[i+1][i][k]+=f[i][j][k])%=p,(f[i+1][i][j]+=f[i][j][k])%=p;
    }for(int i=0;i<N;++i)for(int j=0;j<=(i>0?i-1:i);++j)(ans+=f[N][i][j])%=p;
    printf("%lld",ans*3%p);
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章