[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;
}