Codeforces 1079C Playing Piano（记忆化搜索）

题目链接：Playing Piano

题意：给定长度为n的序列$a_i$，要求得到同样长度的序列$b_i$。满足一下条件：

if $a_i < a_{i+1}$ then $b_i < b_{i+1}$

if $a_i > a_{i+1}$ then $b_i > b_{i+1}$

if $a_i = a_{i+1}$ then $b_i \neq\ b_{i+1}$

$1 <= b_i <= 5$

题解：第一个数从1-5枚举，接着从下一位开始搜索，dp[i][j]表示第i位数字为j的情况是否已经搜索过。

当$a_i<a_{i+1}$的时候，从当前值往后搜索；当$a_i>a_{i+1}$的时候，从当前值往前搜索；$a_i=a_{i+1}$的时候，搜索和当前值不相同的值。搜索的区间范围为1-5。

 #include <cstdio>
 #include <cstdlib>
 using namespace std;

 const int N=1e5+;
 int n,a[N],b[N],dp[N][];

 void dfs(int x,int val){
     if(x>n){
         printf("%d",b[]);
         for(int i=;i<=n;i++) printf(" %d",b[i]);
         printf("\n");
         exit();
     }
     if(dp[x][val]) return ;
     dp[x][val]=;
     if(a[x]>a[x-]){
         for(int i=val+;i<=;i++){
             b[x]=i;
             dfs(x+,i);
         }
     }
     else if(a[x]<a[x-]){
         for(int i=;i<=val-;i++){
             b[x]=i;
             dfs(x+,i);
         }
     }
     else{
         for(int i=;i<=;i++){
             if(i!=val){
                 b[x]=i;
                 dfs(x+,i);
             }
         }
     }
 }

 int main(){
     scanf("%d",&n);
     for(int i=;i<=n;i++) scanf("%d",&a[i]);
     for(int i=;i<=;i++){
         b[]=i;
         dfs(,i);
     }
     printf("-1\n");
     return ;
 }