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();
	}
}