CF 1079 C Playing Piano (Round #522 Div. 2)

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/fufck/article/details/84258204

补坑：两个小时写的构造法 

//#include<iostream>
//#include<cmath>
//#include<cstdlib>
//#include<cstdio>
//#include<algorithm>
#include<bits\stdc++.h>
using namespace std;
int n,a[101010],f[101010][10],pre[101010][10],ans[101010],len=0;
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int i=1;i<=5;i++) f[1][i]=1;
	for(int i=2;i<=n;i++)
	{
		if(a[i]>a[i-1])
		{
			for(int j=1;j<=5;j++)
				for(int k=1;k<=j-1;k++)
				if(f[i-1][k])
				{
					pre[i][j]=k;
					f[i][j]|=f[i-1][k];
				}
		}
		else
		if(a[i]<a[i-1])
		{
			for(int j=1;j<=5;j++)
				for(int k=j+1;k<=5;k++)
				if(f[i-1][k])
				{
					pre[i][j]=k;
					f[i][j]|=f[i-1][k];
				}	
		}
		else
		if(a[i]==a[i-1])
		{
			for(int j=1;j<=5;j++)
				for(int k=1;k<=5;k++)
				if(j!=k&&f[i-1][k])
				{
					pre[i][j]=k;
					f[i][j]|=f[i-1][k];
				}
		}
	}
	int fl=1,ss;
	for(int i=1;i<=5;i++)
		if(f[n][i]==1) {fl=0;ss=i;break;}
	if(fl==1) {printf("-1\n");return 0;}
	ans[++len]=ss;
	int j=n;
	while(pre[j][ss])
	{
		ans[++len]=pre[j][ss];
		ss=pre[j][ss];j--;	
	}
	for(int i=len;i>=1;i--) printf("%d ",ans[i]);
}