【劍指offer】【九度oj】棧的壓入壓出

題目描述：

輸入兩個整數序列，第一個序列表示棧的壓入順序，請判斷第二個序列是否為該棧的彈出順序。假設壓入棧的所有數字均不相等。例如序列1,2,3,4,5是某棧的壓入順序，序列4,5,3,2,1是該壓棧序列對應的一個彈出序列，但4,3,5,1,2就不可能是該壓棧序列的彈出序列。

輸入：

每個測試案例包括3行：

第一行為1個整數n(1<=n<=100000)，表示序列的長度。

第二行包含n個整數，表示棧的壓入順序。

第三行包含n個整數，表示棧的彈出順序。

輸出：

對應每個測試案例，如果第二個序列是第一個序列的彈出序列輸出Yes，否則輸出No。

範例輸入：

5

1 2 3 4 5

4 5 3 2 1

5

1 2 3 4 5

4 3 5 1 2

範例輸出：

Yes

No

大體思路：我們可以建立一個輔助棧來儲存入棧的部分序列，可以o(n)的時間複雜度內解決問題。出棧序列的元素首先與輔助棧（若不空）的棧頂元素比較，若不等再與入棧序列的元素比較。若與入棧序列的元素仍不相等，則當前入棧序列的元素進入輔助棧，出棧序列的當前元素與入棧序列的後面元素繼續比較。若輔助棧的棧頂元素和入棧序列的全部元素都不與出棧序列的當前元素相等，則該序列不是一個正確的出棧序列；若所有出棧序列元素都比較完，則是一個正確的出棧序列。

代碼如下：

#include <iostream>#include <algorithm>#include <string>#include <stack>using namespace std;int a[100010],b[100010];stack<int> s;int main(){int n,i,j;while(cin>>n){for( i=0; i<n; i++ )cin>>a[i];for( i=0; i<n; i++ )cin>>b[i];i=j=0;while( !s.empty() )  //這個很關鍵，因為有多個測試案例，所以需要清空上個測試案例的棧。s.pop();while(i<=n){if( !s.empty() && b[j]==s.top() ){s.pop();j++;}else{if( i==n )break;if( b[j]==a[i] ){i++;j++;}else{s.push(a[i]);i++;}}}if( j == n )cout<<"Yes"<<endl;elsecout<<"No"<<endl;}return 0;}