codeforces round#522-C - Playing Piano 贪心

 

 

贪心写法，看代码好了，虽然用的三目运算有点多

/*疯狂压缩代码行数，在被打的边缘疯狂试探。GIF，=-=*/

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5+7;
int a[maxn],b[maxn],n;
int main()
{
    scanf("%d",&n);a[n]=maxn;
    for(int i=0; i<n; i++)
        scanf("%d",&a[i]);
    b[0]= a[0]<a[1]?1:(a[0]==a[1]?3:5);///贪心第一个为1,3,5
    for(int i=1;i<n;i++){
        if(a[i]==a[i-1])
            if(a[i]==a[i+1])
                b[i] = b[i-1]!=3?3:2;///上一个不是3那就贪3，否则4和2都可以
            else if(a[i]<a[i+1])
                b[i] = b[i-1]!=1?1:2;///明显是一个谷底，贪最小的
            else
                b[i] = b[i-1]!=5?5:4;///山头，贪最大
        else if(a[i]>a[i-1])
            b[i] = (a[i]>a[i+1]&&b[i-1]!=5)?5:b[i-1]+1;
///如果是山头并且没到顶(5),那就贪5，不然两种情况，
///一种是递增(a[i]<=a[i+1])的那就贪花费少，也就是+1
///另一种是山头到顶，一样+1变6被排除
        else
            b[i] = (a[i]<a[i+1]&&b[i-1]!=1)?1:b[i-1]-1;///同上，谷底
        if(b[i]<1||b[i]>5){///超出范围
            printf("-1\n");
            return 0;
        }
    }
    for(int i=0; i<n; i++)
        printf("%d ",b[i]);
    return 0;
}