1022 Train Problem I 棧的簡單應用

題意

棧的簡單應用，給出元素個數，入棧順序O1和出棧順序O2，每個序列裏有n個元素，問能不能利用棧把O1變成O2。

版本1：

依此考慮出棧的元素（記爲temp），首先判斷爲了temp出棧，是否需要有元素入棧，一種是需要，則入棧，入棧之後的棧頂元素則爲temp；若不需要入棧，那麼因爲棧的特點，在正常情況下，棧頂元素爲temp，若不是，則O1不能轉換成O2，此時則退出循環。而對於棧頂元素爲temp的情況，則出棧，繼續考慮下一個出棧元素。

#include<iostream>
#include<stack>
using namespace std;

#define N 10
// 入棧和出棧序列
char in[N];
char out[N];
int main(){
    int x;
    while(~scanf("%d%s%s", &x, in, out)){
        int res[N*2]; // 用於記錄輸入、輸出的過程，入棧爲1，出棧爲0
        int point = 0; // 指向res
        int flag = 1; // 能否轉換的標記
        stack <char> stk;
        // 上次入棧的元素的後面元素的下標
        int start = 0;
        // 判斷每個輸出的元素能否正確輸出
        for (int i = 0; i < x; ++i) {
            char temp;
            temp = out[i];
            // 確定當前輸出元素，在輸入時的下標
            int j = 0;
            for (; j < x; ++j) {
                // 因爲元素是唯一的，所以可以通過判斷元素相等來確定位置
                if(in[j] == temp)
                    break;
            }
            // 確定本輪從哪裏開始入棧
            int k = 0;
            if(j<start)//不需要入棧
            {
                // 是棧頂，出棧
                if(stk.top() == temp){
                    stk.pop();
                    res[point ++] = 1;
                }
                // 出錯
                else{
                    flag = 0;
                    break;
                }
            }
            // 需要入棧
            else{
                // 從上次入棧的位置之後入棧
                k = start;
                for (; k <= j; ++k) {
                    stk.push(in[k]);
                    res[point ++] = 0;
                }
                // 棧頂元素直接出棧
                stk.pop();
                res[point ++] = 1;
                start = ++j;
            }
        }

        // 輸出結果
        if(flag != 0){
            printf("Yes.\n");
            for(int i = 0; i < point; i ++){
                if(res[i] == 1){
                    printf("out\n");
                }
                else
                    printf("in\n");
            }
        }
        else
            printf("No.\n");
        printf("FINISH\n");
    }
    return 0;
}

出錯點：

1. 字符串的輸入。若非一次接收out，而是單獨獲取字符元素，若上次執行失敗（還有元素未接收），會影響下次的輸入。造成Time Limit Exceeded
2. 內存空間的分配。記錄入棧和出棧順序的數組的大小應該是入棧或出棧序列長度的二倍。造成|Memory Limit Exceeded
3. 輸出格式，是‘Yes.’和‘No.’而不是YES.和NO. 造成**Wrong Answer

版本2

參考： 杭電OJ–1021 Train Problem I
熟練使用入棧出棧的模板，先入棧，且在入棧的同時判斷是否要出棧。

#include<iostream>
#include<stack>
#include<vector>
using namespace std;

#define N 10
// 入棧和出棧序列
char in[N];
char out[N];
int main(){
    int x;
    while(~scanf("%d%s%s", &x, in, out)){
        stack <char> stk;
        vector<string> pr;//建立string類型的數組
        // 指向入棧和出棧元素
        int point_in = 0;
        int point_out = 0;
        while(point_in < x)
        {
            // 入棧
            stk.push(in[point_in]);
            point_in ++;
            pr.push_back("in");
            // 出棧
            while(!stk.empty() && out[point_out] == stk.top()){
                stk.pop();
                pr.push_back("out");
                point_out ++;
            }
        }
        // 輸出結果
        if(stk.empty()){
            printf("Yes.\n");
            int len = pr.size();
            for(int i = 0; i < len; i ++){
                // 將pr[i]以C語言形式輸出，或者cout<<pr[i]<<endl;
                printf("%s\n", pr[i].c_str());
            }
        }
        else
            printf("No.\n");
        printf("FINISH\n");
    }
    return 0;
}