ARC 074E RGB Sequence DP

点击打开链接

题意:n件物品,m个要求,3种颜色,n,m<=300,限制为(li,ri,xi) 要求[li~ri]内不同颜色个数为xi 问满足m条件的上色方案数? 

dp[r][g][b] 前k个块中,k=max(r,g,b),最后一个上红色,绿色,蓝色的位置分别为r,g,b &&满足ri<=k的所有条件 

若当前状态合法,则枚举下一个位置的颜色,转移状态.非法的状态dp值设为0, 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> ii;
const int N=3e2+5;
const int M=3e2+5;
const ll mod=1e9+7;
ll dp[N][N][N];//dp[r][g][b] 前k个块中,k=max(r,g,b),最后一个上红色,绿色,蓝色的位置分别为r,g,b &&满足ri<=k的所有条件 
int n,m;
vector<ii> v[N];// 
//当前dp(r,g,b)若非0 则表示是由前k-1个上色后转移来的,所以只要判断ri==k的情况是否合法即可 
bool check(int r,int g,int b)
{
	int k=max(r,max(g,b));
	for(int j=0;j<v[k].size();j++)
	{
		int l=v[k][j].first,x=v[k][j].second;
		//[li~k] 不同数量是否为x 
		int cnt=0;
		if(r>=l)	cnt++;
		if(g>=l)	cnt++;
		if(b>=l) 	cnt++;
		if(cnt!=x)
			return false;
	}
	return true;	
}
int main()
{
	while(cin>>n>>m)
	{
		for(int i=1;i<=n;i++)
			v[i].clear();
		int l,r,x;
		for(int i=1;i<=m;i++)
		{
			cin>>l>>r>>x;
			v[r].push_back(ii(l,x)); 
		}		
		memset(dp,0,sizeof(dp));
		dp[0][0][0]=1;
		ll ans=0;
		for(int r=0;r<=n;r++)
		{
			for(int g=0;g<=n;g++)
			{
				for(int b=0;b<=n;b++)
				{
					if(dp[r][g][b]==0)//非法 
						continue;	
					if(!check(r,g,b))
					{
						dp[r][g][b]=0;
						continue;
					}
					int k=max(r,max(g,b))+1;				
					if(k==n+1)
						ans=(ans+dp[r][g][b])%mod;
					//若当前状态合法则,则枚举下一个位置颜色
					//只有状态都合法 
					dp[k][g][b]+=dp[r][g][b]%mod;
					dp[r][k][b]+=dp[r][g][b]%mod;
					dp[r][g][k]+=dp[r][g][b]%mod;	
				}	
			}	
		}	
		cout<<ans<<endl;	
	}
	return 0;
}