题目链接:https://codeforces.com/contest/1092/problem/D2
题意:跟上道题差不多,给你n个ai,ai表示第i堵墙的高度,现在给一个1*2的砖头,只能横放,不能竖放。问:能否把这片墙砌成高度一直的墙?
题解:也差不多跟上道题差不多,上题是相邻为2的倍数就可以移去,而这题因为只能横放,故相邻为相等才能移去。
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stack>
using namespace std;
int main()
{
int n;
scanf("%d",&n);
stack<int> s;
int MAX=0, a;
bool flag=1;
for(int i=1;i<=n;i++)
{
scanf("%d",&a);
MAX=max(MAX,a);///记录最终砌墙的总高度
if(s.size()==0) s.push(a);
else{
if(s.top()>a) s.push(a);
else if(s.top()==a) s.pop(); ///相邻的墙高度相等才能移去。
else {
flag=0;break;
}
}
}
// printf("flag=%d,MAX=%d %d\n",flag,MAX,s.top());
if(flag==0){
puts("NO");
}
else{
///如果stack里面元素大于一个,故无论怎么横放也无济于事,
///如果stack里面元素恰好一个,这时要元素值要是不等于MAX,也不行,因为我们无法使其它的墙降低高度来屈就于它
if(s.size()>1||(s.size()==1&&s.top()!=MAX)) puts("NO");
else puts("YES");
}
return 0;
}