[HPU] LianLianKan [STL&stack]

题目描述

Nanae很喜欢和朋友玩游戏。虽然有时候看起来很无趣，今天Nanae发明一个新游戏叫LianLianKan。

这个游戏是在一个数字栈玩的，现在有一个数字栈，我们必须从顶部到底部连接和消除相同的一对元素。

你只能连接和消除与栈顶元素相同的元素，从栈里消除他们之后，所有的元素会下落。尽管这游戏好像非常有趣，但是事实上它真的非常无聊。

所以，为了证明我在Nanae的朋友里面是一个聪明人，Nanae加多了一条规则在游戏里面：对于每一个栈顶元素，它只能与距离小于6的相同的值元素连接和消除。在玩游戏之前，他想检查我是否能够消除数字栈里的所有元素

（注：两个元素之间的距离是1.）

输入

第一行是一个整数N，表示数字栈有多少个元素。(1 <= N <= 1000)

下一行有N个数，表示数字栈里面从底部到顶部有N个整数ai。(0 <= ai <= 2,000,000,000)

输出

对于每一个案例，输出“1”表示能消除所有元素，否则输出0。

样例输入

2

1 1

3

1 1 1

2

1000000 1

样例输出

1

0

0

#include<stdio.h>
#include<stack>
#include<queue>
#define MAX_N 1002
using namespace std;
int a[MAX_N];

int main()
{
int N;
while(scanf("%d",&N)!=EOF){
stack<int> S;
stack<int> tmp;
for(int i=0;i<N;i++)
scanf("%d",&a[i]);

for(int i=0;i<N;i++){
int cnt=6;bool flag=true;
while(cnt--){
if(S.empty()){
flag=false;
break;
}

int t=S.top();S.pop();
if(t==a[i]) break;
tmp.push(t);
if(cnt==0) flag=false;
}

while(!tmp.empty()){
S.push(tmp.top());
tmp.pop();
}

if(!flag) S.push(a[i]);
}
puts(S.empty()?"1":"0");
}
return 0;
}