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

题意：给出一个数列 a1 a2......an  让你构造一个序列(该序列取值（1-5）)

如果a(i+1)>a(i) b(i+1)>b(i)

如果a(i+1)<a(i)  那么b(i+1)<b(i)

如果a(i+1)==a(i) 那么b(i+1)!=b(i)

请构造该序列 如果不存在 那么输出-1

思路： 当时想的时候是贪心 但是贪心的话对于相等的时候就很不好处理

然而该题的状态比较好定义 dp[i][x] 表示 第i个位置 取x 是否成立

那么状态总量有  n(<=5e5)*5 也就是 25e*5 直接记忆化搜索即可

 #include<bits/stdc++.h>
 #define FOR(i,f_start,f_end) for(int i=f_start;i<=f_end;i++)
 #define MS(arr,arr_value) memset(arr,arr_value,sizeof(arr))
 #define F first
 #define S second
 #define pii pair<int ,int >
 #define mkp make_pair
 #define pb push_back
 #define arr(zzz) array<ll,zzz>
 #define ll long long
 using namespace std;
 const int maxn=1e6+;
 int a[maxn];
 int vis[maxn][];
 int n;
 int ans[maxn];
 bool dfs(int x){//5*(5e5)
     if(x==n)return ;
     for(int i=;i<=;i++){
         if(vis[x+][i])continue;
         if((a[x+]>a[x]&&i>ans[x])||(a[x+]<a[x]&&i<ans[x])||(a[x+]==a[x]&&i!=ans[x])){
             vis[x+][i]=;
             ans[x+]=i;
                 if(dfs(x+))return ;
         }
     }
     return ;
 }

 int main(){
     scanf("%d",&n);
     for(int i=;i<=n;i++)scanf("%d",&a[i]);
     for(int i=;i<=;i++){
         ans[]=i;
         vis[][i]=;
         if(dfs()){
         for(int i=;i<=n;i++)cout<<ans[i]<<" ";
                 return ;
         }

     }
     cout<<-<<endl;
     return ;
 }