codeforces round#522-C - Playing Piano 貪心
阿新 • • 發佈:2018-12-03
貪心寫法,看程式碼好了,雖然用的三目運算有點多
/*瘋狂壓縮程式碼行數,在被打的邊緣瘋狂試探。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; }