[Codeforces Round #522 (Div. 2, based on Technocup 2019 Elimination Round 3)][C. Playing Piano]

http://codeforces.com/contest/1079/problem/C

题目大意:给出一个数列a[n],让构造一个满足下列条件的数列b[n]:如果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].其中1<=b[i]<=5  1<=a[i]<=2*1e5.

题解:dp[i][j]表示在位置i处取j时是否成立,如果成立则dp[i][j]=1,而这个成立的条件就是这个状态之前的状态成立dp[i-1][k]=1,这样就可以O(5*N)解决这个问题(对于这种要求构造出数列并且新数列的取值很少的题目,直接枚举取值利用dp方式做即可)

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
typedef long long ll;
int a[100005],b[100005];
int dp[100005][6],pre[100005][6];
int main(){
 // freopen("in.txt","r",stdin);
    int n;
    int f=1;
    int tot1=0;
    int tot2=0;
    cin>>n;
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    for(int i=1;i<=5;i++){
        dp[1][i]=1;
    }
    for(int i=1;i<n;i++){
        for(int j=1;j<=5;j++){
            if(!dp[i][j])continue;
            if(a[i+1]>a[i]){
                for(int k=j+1;k<=5;k++){
                    dp[i+1][k]=1;
                    pre[i+1][k]=j;
                }
            }
            else if(a[i+1]<a[i]){
                for(int k=1;k<j;k++){
                    dp[i+1][k]=1;
                    pre[i+1][k]=j;
                }
            }
            else{
                for(int k=1;k<=5;k++){
                    if(k==j)continue;
                    dp[i+1][k]=1;
                    pre[i+1][k]=j;
                }
            }
        }
    }
    for(int i=1;i<=5;i++){
        if(dp[n][i]){
            for(int j=n;j>=1;j--){
                b[j]=i;
                i=pre[j][i];
            }
            f=0;
            break;
        }
    }
    if(f){printf("-1
");return 0;}
    for(int i=1;i<=n;i++){
        printf("%d",b[i]);
        char cc=(i==n)?'
':' ';
        printf("%c",cc);
    }
    return 0;
}