1. 程式人生 > >Codeforces Round #527 (Div. 3) D2. Great Vova Wall (Version 2) (棧)

Codeforces Round #527 (Div. 3) D2. Great Vova Wall (Version 2) (棧)

題意:有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");
}