面试题22:栈的压入、弹出序列
题目:输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1、2、3、4、5是某栈的压栈序列,序列4、5、3、2、1是该栈序列对应的一个弹出序列,但4、3、5、1、2就不可能是该压栈序列的弹出序列。
思路:按照题目中的例子,序列1为1、2、3、4、5,序列2为4、5、3、2、1。定义一个辅助栈,按照序列2的出栈顺序入栈。现在4要出栈,就必须把1,2,3入栈。然后4入栈再出栈,5入栈再出栈。3,2,1再出栈
//栈的压入弹出序列
bool IsPopOrder(const int*arr,const int*brr,int len)
{
assert(arr!=NULL && brr!=NULL && len>0);
int i = 0;//第一个序列的下标
int j = 0;//第二个序列的下标
stack<int> dataStack;//辅助栈
while(j < len)
{
//要出栈的元素 之前的元素必须全部入栈
while(dataStack.empty()
|| dataStack.top()!=brr[j])
{
dataStack.push(arr[i]);
i++;
}
if(dataStack.top() != brr[j])
break;
//按照第二个序列出栈
dataStack.pop();
j++;
}
//第二个序列的值已经全部出栈,栈为空
if(dataStack.empty() && j==len)
return true;
else
return false;
}
int main()
{
int arr[] = {1,2,3,4,5};
int brr[] = {4,5,3,2,1};
int crr[] = {4,3,5,1,2};
cout<<IsPopOrder(arr,brr,sizeof(arr)/sizeof(arr[0]))<<endl;
cout<<IsPopOrder(arr,crr,sizeof(arr)/sizeof(arr[0]))<<endl;
return 0;
}