【牛客網】棧的壓入、彈出序列
阿新 • • 發佈:2019-01-28
轉載於牛客網
- 題目:
輸入兩個整數序列,第一個序列表示棧的壓入順序,請判斷第二個序列是否為該棧的彈出順序。假設壓入棧的所有數字均不相等。例如序列1,2,3,4,5是某棧的壓入順序,序列4,5,3,2,1是該壓棧序列對應的一個彈出序列,但4,3,5,1,2就不可能是該壓棧序列的彈出序列。(注意:這兩個序列的長度是相等的)
- AC的程式碼:
class Solution {
public:
bool IsPopOrder(vector<int> pushV,vector<int> popV) {
if (pushV.empty() || popV.empty())return 0;
vector<int>stack;
for (int i = 0, j = 0; i < pushV.size();){
stack.push_back(pushV[i++]);
while (j < popV.size() && stack.back() == popV[j]){
stack.pop_back();
j++;
}
}
return stack.empty();
}
};
- 思路:
借用一個輔助的棧,遍歷壓棧順序,先將pushV中第一個放入棧中,然後判斷棧頂元素是不是出棧順序的第一個元素,如果相等,那麼出棧;如果不等,繼續壓棧,直到相等以後開始出棧,出棧一個元素,則將出棧順序向後移動一位,直到不相等,這樣迴圈等壓棧順序遍歷完成,如果輔助棧還不為空,說明彈出序列不是該棧的彈出順序。