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

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/fufck/article/details/84258852
#include<iostream>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<cstdio>
#include<algorithm>
#include<string>
#include<map>
#include<queue>
#include<vector>
#include<stack>
#define ll long long
#define maxn 400100
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define PI 2*asin(1)
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define SWAP(x,y,t) ( (t)=(x),(x)=(y),(y)=(t) )
using namespace std;
int l=1,n,flag,fl=0,bz;
int q[maxn],a[maxn],p[maxn],f[maxn];
int main()
{
//	freopen("1.txt","r",stdin);
//	freopen("1.out","w",stdout); 
	scanf("%d",&n);a[0]=-1;
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	if(a[1]==a[2]) flag=1;
	if(a[1]<a[2]) flag=2;
	if(a[1]>a[2]) flag=3;//1表示平的 2表示上升 3表示下降  
	q[1]=1;p[1]=flag;//q记录的是连续的区间,p记录是1,2,还是3 
	for(int i=3;i<=n;i++)
	{
		if(a[i-1]>a[i]&&flag!=3)
		{
			p[++l]=3;
			q[l]=i-1;
			flag=3;
		}
		else{
			if(a[i-1]<a[i]&&flag!=2)
			{
				p[++l]=2;
				q[l]=i-1;
				flag=2;
			}
			else {
				if(a[i-1]==a[i]&&flag!=1)
				{
					p[++l]=1;
					q[l]=i-1;
					flag=1;
				}
			}
		}
	}
	q[l+1]=n;
	for(int i=1;i<=l;i++)
	{
		if(p[i]==2&&p[i+1]==3)
		{
			if(f[q[i]]==0) f[q[i]]=1;
			for(int j=q[i]+1;j<=q[i+1];j++) f[j]=f[j-1]+1;
			f[q[i+1]]=max(5,f[q[i+1]]);
			if(f[q[i+1]]>5) fl=1;
		}//如果是先上升再下降的情况,那个最高点的值就是 5,对后面的帮助大 
		if(p[i]==2&&p[i+1]!=3)
		{
			if(f[q[i]]==0) f[q[i]]=1;
			for(int j=q[i]+1;j<=q[i+1];j++) f[j]=f[j-1]+1;	
			if(f[q[i+1]]>5) fl=1;
		}//如果是先上升再平的情况,那个最高点的值累加就行,平的时候可以随意调节数值 
		if(p[i]==3&&p[i+1]==2)
		{
			if(f[q[i]]==0) f[q[i]]=5;
			for(int j=q[i]+1;j<=q[i+1];j++) f[j]=f[j-1]-1;
			f[q[i+1]]=min(1,f[q[i+1]]);
			if(f[q[i+1]]<1) fl=1;
		}
		if(p[i]==3&&p[i+1]!=2)
		{
			if(f[q[i]]==0) f[q[i]]=5;
			for(int j=q[i]+1;j<=q[i+1];j++) f[j]=f[j-1]-1;	
			if(f[q[i+1]]<1) fl=1;
		}
		if(p[i]==1)
		{
			if(p[i+1]==2) bz=1;
			else bz=5; 
			for(int k=q[i];k<=q[i+1]-1;k++)
			{
				if(f[k]) continue;
				for(int j=1;j<=5;j++)
				   if(j!=bz&&j!=f[k-1]) 
				   {
				   	 f[k]=j;break;
				   }
			}//平的一段的值只要不能与后面那段的最大或者最小值就行,最后一个数要么是5,要么是1。 
			if(q[i+1]-q[i]==1&&q[i]!=1)//特判两个数平的时候最后那个数一定要和前面的不一样,所以只能直接赋值 
			{
				int k=q[i+1];
				  if(bz==1)
				  {
					for(int j=1;j<=5;j++)
					   if(j!=f[k-1]) 
					   {
					   	 f[k]=j;break;
					   }
			      }
			      else
			      {
			      	for(int j=5;j>=1;j--)
					   if(j!=f[k-1]) 
					   {
					   	 f[k]=j;break;
					   }
				  }
			}
		}
	}
	if(fl) 
	{
		printf("-1");
		return 0;
    }
    if(f[n]==0)
    {
    	for(int j=1;j<=5;j++) if(j!=f[n-1]) { f[n]=j;break;}
	}
    for(int i=1;i<=n;i++) printf("%d ",f[i]);
} 

猜你喜欢

转载自blog.csdn.net/fufck/article/details/84258852