Codeforces Round #522 div 2 C. Playing Piano dp

C. Playing Piano

题意：给你一个a数组，你要按要求构造等长的b数组，如果a[ i ]>a[ i-1 ]，那么b[ i ]>b[ i-1 ]，若a[ i ]<a[ i-1]，那么b[ i ]<b[ i-1 ]，若a[ i ]=a[ i-1 ]，那么b[ i ]！=b[ i-1 ]，且b数组最大不超过5，最小为1。

思路：一个很有意思的题，设d[ i ][ j ][ k ]表示b[ i ]=j 且b[ i-1 ]=k时，是否合理，状态转移很简单，初始化d[ 1 ][ j ][ k ]全为1，若a[ i ]>a[ i-1 ]，那么d[ i ][ j ][ k ] |= d[ i-1 ][ k ][ p ]（j>k，1<=p<=5），就ok了（欢迎大家优化这个dp）。

#include<cstdio>
#include<cstring>
#include<stack>
using namespace std;
const int maxn=1e5+10;
int d[maxn][6][6],a[maxn];
stack<int>s;
int main()
{
	int n,x;
	scanf("%d%d",&n,&a[1]);
	if(n==1)
	{
		puts("1");
		return 0;
	}
	memset(d[1],1,sizeof(d[1]));
	for(int i=2;i<=n;i++)
	{
		scanf("%d",&a[i]);
		for(int j=1;j<=5;j++)
		for(int k=1;k<=5;k++)
		for(int p=1;p<=5;p++)
		{
			if(a[i]==a[i-1])
			{
				if(j!=k)
				d[i][j][k]|=d[i-1][k][p];
			}
			else if(a[i]>a[i-1])
			{
				if(j>k)
				d[i][j][k]|=d[i-1][k][p];
			}
			else if(a[i]<a[i-1])
			{
				if(j<k)
				d[i][j][k]|=d[i-1][k][p];
			}
		}
	}
	int p=0;
	for(int j=1;j<=5;j++)
	for(int k=1;k<=5;k++)
	if(d[n][j][k])p=j;
	if(!p)puts("-1");
	else
	{
		for(int i=n;i;i--)
		{
			s.push(p);
			for(int j=1;j<=5;j++)
			if(d[i][p][j])
			{
				p=j;
				break;
			}
		}
		while(!s.empty())
		printf("%d ",s.top()),s.pop();
	}
}