1. 程式人生 > >codeforces round#522-C - Playing Piano 貪心

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;
}