面試題之【用兩個棧實現佇列】

林堯彬發表於2020-04-04

題目很簡單,與之相似的還有用兩個佇列實現棧,思路類似都是用一個村一個倒,類似負負得正嘛。

具體分析一下兩個棧實現佇列,設這兩個分別為s1和s2,我們從入隊開始,最開始只要直接壓倒s1中,然後出隊,此事要先將元素全部彈到出再放到s2中;現在的問題是當兩個棧都有東西的時候要怎麼處理,其實分析一下我們發現s2中的元素就是最先進的,所以pop只要彈s2就行,同理壓棧只要壓到s1裡面,程式碼如下:

 1 #include<iostream>
 2 #include<cstdio>
 3 #include<stack>
 4 using namespace std;
 5 const std::string PUSH="PUSH";
 6 const std::string POP="POP";
 7 int main()
 8 {
 9     int n;
10     while(scanf("%d",&n)==1)
11     {
12         std::stack<int> s1;
13         std::stack<int> s2;
14         for(int i=0;i<n;i++)
15         {
16             std:string a;
17             int num;
18             std::cin>>a;
19             if(a==PUSH)
20             {
21                 cin>>num;
22                 s1.push(num);
23             }
24             else
25             {
26                 if(!s2.empty())
27                 {
28                     printf("%d\n",s2.top());
29                     s2.pop();
30                 }
31                 else
32                 {
33                     while(!s1.empty())
34                     {
35                         s2.push(s1.top());
36                         s1.pop();
37                     }
38                     if(!s2.empty())
39                     {
40                         printf("%d\n",s2.top());
41                         s2.pop();
42                     }
43                     else
44                     {
45                         printf("-1\n");
46                     }
47                 }
48             }
49         }
50     }
51     return 0;
52 }

 

轉載於:https://www.cnblogs.com/MrLJC/p/3646412.html

相關文章