1. 題目

棧排序。 編寫程序，對棧進行排序使最小元素位于棧頂。
最多只能使用一個其他的臨時棧存放數據，但不得將元素復制到別的數據結構（如數組）中。
該棧支持如下操作：push、pop、peek 和 isEmpty。當棧為空時，peek 返回 -1。

示例1:輸入： ["SortedStack", "push", "push", "peek", "pop", "peek"] [[], [1], [2], [], [], []]輸出： [null,null,null,1,null,2]示例2:輸入： ["SortedStack", "pop", "pop", "push", "pop", "isEmpty"] [[], [], [], [1], [], []]輸出： [null,null,null,null,null,true] 說明: 棧中的元素數目在[0, 5000]范圍內。

2. 解題

• 用兩個棧，來回倒數據，保證其中一個為空
• 倒的過程中找到最小的放在非空的棧的頂端

class SortedStack {stack<int> s;stack<int> temp;int nextMin; public:SortedStack() {}void push(int val) {if(isEmpty())s.push(val);else if(temp.empty()){if(s.top() < val)//棧頂小swap(val,s.top());//把棧頂變成大的s.push(val);//在把原來的小棧頂放回}else //s.empty(){if(temp.top() < val)swap(val,temp.top());temp.push(val);}}void pop() {if(isEmpty())return;if(temp.empty()){s.pop();//最小值popwhile(!s.empty())//倒數據到temp{nextMin = s.top();s.pop();if(!temp.empty() && nextMin > temp.top())swap(nextMin, temp.top());temp.push(nextMin);}}else //s.empty(){temp.pop();//最小值while(!temp.empty())//倒數據到s{nextMin = temp.top();temp.pop();if(!s.empty() && nextMin > s.top())swap(nextMin, s.top());s.push(nextMin);}}}int peek() {if(isEmpty())return -1;if(temp.empty())return s.top();else //s.empty()return temp.top();}bool isEmpty() {return s.empty()&&temp.empty();} };

總結

以上是生活随笔為你收集整理的程序员面试金典 - 面试题 03.05. 栈排序（两栈）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。