牛客练习赛96-小y的容器-(dp+思维转化)

E

题意：
就是给你n个数，然后3个容器，每次你可以把数放在某个容器中。最后每个容器必须要有一个数，然后每个容器的值从小到大排序后相邻两个的差不能超过3。问你所有的合法方案。

思考：
其实看到感觉很复杂，这怎么弄。只知道不能超过三，那么可以这样定义dp[i][a][b][c],走到第i位，第一个容器的最后一个是a，第二个容器的最后一个是b，第三个容器的最后一个是c。很明显直接从a-1,a-2,a-3…来转移，但是这样炸空间了。其实仔细看看，由于就是递增的去放，每次的值都会变大1，所以定义dp[i][a][b][c]，代表走到i了，当前和i的差值是a的方案数。那么这个状态就可以去推i+1的状态，只要a+1<=3，为什么a+1呢因为i值变大了1所以差值也只变大1。0 1 2 3就是正常的差值，4代表当前这个容器为空的时候。所以这题用推标法比较好做。

代码：

int T,n,m,k;
int va[N][5];
int dp[N][5][5][5];

int f(int x)
{
	if(x==3||x==4) return x;
	else return x+1;
}

signed main()
{
	IOS;
	cin>>n>>m;
	for(int i=1;i<=m;i++)
	{
		int a,b;
		cin>>a>>b;
		va[a][b] = 1;
	}
	dp[0][4][4][4] = 1;
	for(int i=1;i<=n;i++)
	{
		for(int a=0;a<=4;a++)
		{
			for(int b=0;b<=4;b++)
			{
				for(int c=0;c<=4;c++)
				{
					if(!va[i][1]&&(a==4||a+1<=3))
					dp[i][0][f(b)][f(c)] = (dp[i][0][f(b)][f(c)]+dp[i-1][a][b][c])%mod;
					if(!va[i][2]&&(b==4||b+1<=3))
					dp[i][f(a)][0][f(c)] = (dp[i][f(a)][0][f(c)]+dp[i-1][a][b][c])%mod;
					if(!va[i][3]&&(c==4||c+1<=3))
					dp[i][f(a)][f(b)][0] = (dp[i][f(a)][f(b)][0]+dp[i-1][a][b][c])%mod;
				}
			}
		}
	}
	int ans = 0;
	for(int a=0;a<=3;a++)
	{
		for(int b=0;b<=3;b++)
		{
			for(int c=0;c<=3;c++)
			ans = (ans+dp[n][a][b][c])%mod;
		}
	}
	cout<<ans;
	return 0;
}

总结：
多多思考，多多积累经验。