Codeforces Round #527 (Div. 3) D2. Great Vova Wall (Version 2) (棧)
阿新 • • 發佈:2018-12-24
題意:有n個列,然後輸入n個數ai表示每個列當前的磚的個數,然後只有有1*2的磚,問最後能不能鋪滿n*max(ai)
思路:由於只能用1*2的磚,所以只有兩塊磚的高度相同時才可以消去,還有就是像這樣的1221不滿足題意,因為兩邊的相同的高度沒法合在一起,像這樣的2112這種才滿足,最後可以有一列高度為h的牆,不過h要滿足h >= max( h ) 才可以,用棧模擬過程。
#include<bits/stdc++.h> using namespace std; #define read(x) scanf("%d",&x) int main() { int n, h; read(n); int mx = -1; stack <int> sta; for(int i = 0; i < n; i++) { read(h); mx = max(mx,h); if(!sta.empty()) { if(sta.top() < h) return puts("NO"), 0; if(sta.top() == h) sta.pop(); else sta.push(h); } else sta.push(h); } if(sta.size() > 1) puts("NO"); else if(sta.size() == 1 && sta.top() < mx) puts("NO"); else puts("YES"); }