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

Codeforces Round #527 (Div. 3) D1. Great Vova Wall (Version 1)

題目:

傳送門

題意:
給出一個2*1的磚塊,有兩種放磚塊的操作,一種是橫著放,這樣會使相鄰的兩塊牆高度加1,一種是豎著放,這樣會使牆高度加2,問以上兩種操作是否可以使建成的牆高度相同,並且建成的牆沒有空隙。

思路:
因為每塊牆可以通過無限次數的豎著放達到最接近相鄰牆的高度,所以真正影響牆的建成的其實是橫著放,設想若有兩個牆初始高度相同,那麼與它相鄰的兩塊牆的奇偶性相同的話,則一定可以建成一樣高度的牆。

所以我們可以通過建立一個棧實現模擬操作,如果與棧頂元素不相同,則push進一個元素,若相同,則刪除棧頂元素。

最後如果棧中的元素小於2,則可以建成,否則,不能建成。

程式碼如下:

#include <bits/stdc++.h>
using namespace std;
const int maxn=2*1e5+5;
int a[maxn];
int n;
int main()
{
    scanf("%d",&n);
    stack<int>s;
    for (int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
        a[i]%=2;
        if(!s.empty()&&s.top()==a[i])
        {
            s.pop();
        }
        else
        {
            s.push(a[i]);
        }
    }
    if(s.size()<2)
    {
        printf("YES\n");
    }
    else
    {
        printf("NO\n");
    }
    return 0;
}