问题描述:
输入两个整数序列,第一个表示栈的压入顺序,判断第二个序列是否为该栈的弹出序列(假设压栈的所有数字均不相等)。
例如:序列1,2,3,4,5是压栈序列,那么序列4,5,3,2,1就是一个合法的弹出序列,而4,3,5,1,2则不可能是合法的弹出序列。
问题出自《剑指offer》面试题22。
解题代码:
思路和书中讲诉类似:如果下一个弹出元素正好是栈顶元素,则直接弹出。否则把压栈序列中还没有入栈的元素依次入栈,直到栈顶元素等于弹出序列中要弹出的元素。如果所用元素都压栈了,并且栈顶元素任然不等于要弹出元素,则不合法。
代码实现上和书中有所区别:用两个下标i,j分别扫描压栈序列和弹出序列,循环的退出条件是:j < n && (i < n || !sta.empty())。退出循环时,如果弹出序列扫描完毕,则该弹出序列是合法的,否则不合法。
代码注意两点:
1,当i == n时,如果栈不为空,循环不能退出,还要继续执行循环体;
2,判断当前出栈元素是不是当前栈顶元素之前,要判断栈是否为空。如不判断,在栈为空时,执行取栈顶元素,程序异常。
代码如下:
#include <stdio.h>
#include <stdlib.h>
#include <stack>
using namespace std;
bool judge(int *a, int *b, int n) {
stack<int> sta;
int i = 1, j = 0;
sta.push(a[0]);
while (j < n && (i < n || !sta.empty())) {
if (!sta.empty() && b[j] == sta.top()) {
sta.pop();
j++;
} else {
if (i < n) {
sta.push(a[i]);
i++;
} else {
return false;
}
}
}
if (j == n) return true;
else return false;
}
int main() {
int n, i;
while (1) {
scanf("%d", &n);
int *a = new int[n];
for (i = 0; i < n; i++) {
scanf("%d", &a[i]);
}
int *b = new int[n];
for (i = 0; i < n; i++) {
scanf("%d", &b[i]);
}
bool rslt = judge(a, b, n);
printf("%d\n", rslt);
}
}