CodeForces Round #527 (Div3) D2. Great Vova Wall (Version 2)
http://codeforces.com/contest/1092/problem/D2
Vova's family is building the Great Vova Wall (named by Vova himself). Vova's parents, grandparents, grand-grandparents contributed to it. Now it's totally up to Vova to put the finishing touches.
The current state of the wall can be respresented by a sequence aa of nn integers, with aiai being the height of the ii-th part of the wall.
Vova can only use 2×12×1 bricks to put in the wall (he has infinite supply of them, however).
Vova can put bricks only horizontally on the neighbouring parts of the wall of equal height. It means that if for some ii the current height of part ii is the same as for part i+1i+1, then Vova can put a brick there and thus increase both heights by 1. Obviously, Vova can't put bricks in such a way that its parts turn out to be off the borders (to the left of part 11 of the wall or to the right of part nn of it).
Note that Vova can't put bricks vertically.
Vova is a perfectionist, so he considers the wall completed when:
- all parts of the wall has the same height;
- the wall has no empty spaces inside it.
Can Vova complete the wall using any amount of bricks (possibly zero)?
The first line contains a single integer nn (1≤n≤2⋅1051≤n≤2⋅105) — the number of parts in the wall.
The second line contains nn integers a1,a2,…,ana1,a2,…,an (1≤ai≤1091≤ai≤109) — the initial heights of the parts of the wall.
Print "YES" if Vova can complete the wall using any amount of bricks (possibly zero).
Print "NO" otherwise.
5 2 1 1 2 5
YES
3 4 5 3
NO
2 10 10
YES
In the first example Vova can put a brick on parts 2 and 3 to make the wall [2,2,2,2,5][2,2,2,2,5] and then put 3 bricks on parts 1 and 2 and 3 bricks on parts 3 and 4 to make it [5,5,5,5,5][5,5,5,5,5].
In the second example Vova can put no bricks in the wall.
In the third example the wall is already complete.
代码:
#include <bits/stdc++.h> using namespace std; int N; stack<int> s; int main() { scanf("%d", &N); bool flag = true; int maxx = INT_MIN; for(int i = 0; i < N; i ++) { int x; scanf("%d", &x); maxx = max(maxx, x); if(s.empty()) s.push(x); else { if(s.top() < x) flag = false; else if(s.top() == x) s.pop(); else s.push(x); } } if(s.size() > 1) flag = false; else if(s.size() == 1 && s.top() < maxx) flag = false; if(flag) printf("YES\n"); else printf("NO\n"); return 0; }
emmmm 两个墙的高度如果一样的话可以通过横着放增加到任意高度按照这个运用栈 好像是除了做运算第一次用栈正经的写题