[九度][何海涛] 栈的压入压出

题目描述：

输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序，序列4，5,3,2,1是该压栈序列对应的一个弹出序列，但4,3,5,1,2就不可能是该压栈序列的弹出序列。

 

输入：

每个测试案例包括3行：

第一行为1个整数n(1<=n<=100000)，表示序列的长度。

第二行包含n个整数，表示栈的压入顺序。

第三行包含n个整数，表示栈的弹出顺序。

 

输出：

对应每个测试案例，如果第二个序列是第一个序列的弹出序列输出Yes，否则输出No。

 

样例输入：

5
1 2 3 4 5
4 5 3 2 1
5
1 2 3 4 5
4 3 5 1 2

样例输出：

Yes
No

用一个栈来模拟，可以做到O(n)

#include <iostream>
#include <stack>
using namespace std;

int main()
{
    int a[100000];
    int b[100000];
    int n;
    stack<int> s;

    while(cin >> n)
    {
        for(int i = 0; i < n; i++)
            cin >> a[i];

        for(int i = 0; i < n; i++)
            cin >> b[i];

        bool flag = true;
        int index = 0;
        while(!s.empty())
            s.pop();

        for(int i = 0; i < n; i++)
        {
            if (!s.empty() && s.top() == b[i])
                s.pop();
            else
            {
                bool findIt = false;
                while(index < n)
                {
                    s.push(a[index]);
                    if (a[index] == b[i])
                    {
                        index++;
                        findIt = true;
                        break;
                    }

                    index++;
                }

                if (!findIt)
                {
                    flag = false;
                    break;
                }
                else
                    s.pop();
            }
        }

        if (flag)
            cout << "Yes" << endl;
        else
            cout << "No" << endl;
    }
}