剑指Offer - 九度1366 - 栈的压入、弹出序列

剑指Offer - 九度1366 - 栈的压入、弹出序列
2014-02-05 20:41

题目描述：

输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列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

题意分析：
　　给定一个入栈的数列，判断给定的另一个数列是否可能是一个出栈序列。
　　要做的第一件事，就是将a[1],a[2],...,a[n]和1,2,...,n进行映射。这样方便判断大小，找到一个判断先后顺序的参照标准。
　　接下来，问题就转化成了：如果入栈序列为1,2,...,n，出栈顺序是否可能为a[1],a[2],...,a[n]？
　　具体的思路请直接看代码，几个if分支的意思应该比较明确了。过程中需要一个栈来模拟入栈出栈过程。
　　时间复杂度为O(n)，空间复杂度也是O(n)。

 // 688864    zhuli19901106    1366    Accepted    点击此处查看所有case的执行结果    6504KB    916B    640MS
 //
 #include <cstdio>
 #include <map>
 #include <stack>
 using namespace std;

 map<int, int> m2;
 stack<int> st;
 const int MAXN = ;
 int a[MAXN];

 int main()
 {
     int i, j, n;
     int tmp;

     while (scanf("%d", &n) == ) {
         for (i = ; i < n; ++i) {
             scanf("%d", &tmp);
             m2[tmp] = i;
         }
         for (i = ; i < n; ++i) {
             scanf("%d", &tmp);
             a[i] = m2[tmp];
         }

         i = j = ;
         while (j < n) {
             if (a[j] == i) {
                 ++i;
                 ++j;
             } else if (!st.empty() && st.top() == a[j]) {
                 st.pop();
                 ++j;
             } else if (i < n) {
                 st.push(i);
                 ++i;
             } else {
                 break;
             }
         }
         if (i == n && j == n && st.empty()) {
             printf("Yes\n");
         } else {
             printf("No\n");
         }

         while (!st.empty()) {
             st.pop();
         }
         m2.clear();
     }

     return ;
 }