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

题意：给出一个数列 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 直接记忆化搜索即可

 1 #include<bits/stdc++.h>
 2 #define FOR(i,f_start,f_end) for(int i=f_start;i<=f_end;i++)
 3 #define MS(arr,arr_value) memset(arr,arr_value,sizeof(arr)) 
 4 #define F first 
 5 #define S second
 6 #define pii pair<int ,int >
 7 #define mkp make_pair
 8 #define pb push_back
 9 #define arr(zzz) array<ll,zzz>
10 #define ll long long 
11 using namespace std;
12 const int maxn=1e6+100;
13 int a[maxn];
14 int vis[maxn][10];
15 int n;
16 int ans[maxn];
17 bool dfs(int x){//5*(5e5)
18     if(x==n)return 1;
19     for(int i=1;i<=5;i++){
20         if(vis[x+1][i])continue;
21         if((a[x+1]>a[x]&&i>ans[x])||(a[x+1]<a[x]&&i<ans[x])||(a[x+1]==a[x]&&i!=ans[x])){
22             vis[x+1][i]=1;
23             ans[x+1]=i;
24                 if(dfs(x+1))return 1;
25         }
26     }
27     return 0;
28 }
29 
30 int main(){
31     scanf("%d",&n);
32     for(int i=1;i<=n;i++)scanf("%d",&a[i]);
33     for(int i=1;i<=5;i++){
34         ans[1]=i;
35         vis[1][i]=1;
36         if(dfs(1)){
37         for(int i=1;i<=n;i++)cout<<ans[i]<<" ";
38                 return 0;
39         }
40 
41     }
42     cout<<-1<<endl;
43     return 0;
44 }