題目:
輸入兩個整數序列,第一個序列表示棧的壓入順序,請判斷第二個序列是否為該棧的彈出順序。假設壓入棧的所有數字均不相等。例如,序列 {1,2,3,4,5} 是某棧的壓棧序列,序列 {4,5,3,2,1} 是該壓棧序列對應的一個彈出序列,但 {4,3,5,1,2} 就不可能是該壓棧序列的彈出序列。
示例:
輸入:pushed = [1,2,3,4,5], popped = [4,5,3,2,1]
輸出:true
解釋:我們可以按以下順序執行:
push(1), push(2), push(3), push(4), pop() -> 4,
push(5), pop() -> 5, pop() -> 3, pop() -> 2, pop() -> 1
思路:
用棧來模擬數組的入棧出棧過程。
遍歷pushed數組,遍歷過程中:
把當前元素入棧;
不斷檢查棧頂元素和popped數組的當前元素是否相同,如果相同,則出棧。
最后檢查棧元素是否為空。
代碼:
class Solution {
public:
bool validateStackSequences(vector<int>& pushed, vector<int>& popped) {
if(pushed.size() != popped.size())
return false;
stack<int> stk;
int j=0;
for(int i=0; i<n; i++)
{
stk.push(pushed[i]);
while(!stk.empty() && stk.top() == popped[j])
{
stk.pop();
j++;
}
}
return stk.empty();
}
};