[刷题] PTA 02-线性结构4 Pop Sequence

模拟栈进出

方法一：

#include<stdio.h>
#define MAXSIZE 1000

typedef struct{
    int data[MAXSIZE];
    int top;
}SqStack;

int InitStack(SqStack *s){
    s->top=-1;
    return 0;
}

int Push(SqStack *s,int e){
    if(s->top==MAXSIZE) return 1;
    s->top++;
    s->data[s->top]=e;
    return 0;
}

int Pop(SqStack *s){
    int e;
    if(s->top==-1) return 1;
    e = s->data[s->top];
    s->top--;
    return e;
}

int main(){
    int m,n,k,i,h;
    SqStack s,t;
    scanf("%d%d%d",&m,&n,&k);
    while(k--){
        InitStack(&s);
        InitStack(&t);
        for(i=n-1;i>=0;i--){
            scanf("%d",&(t.data[i]));
        }
        t.top=n-1;
        i=0;
        while(i<=n){
            if(s.data[s.top]==t.data[t.top]&&s.top!=-1){
                Pop(&s);
                Pop(&t);
            }else if(s.top<m-1 && i<n){
                i++;
                Push(&s,i);
            }else{
                break;
            }
        }
        if(s.top==-1 && t.top==-1){
            printf("YES
");
        }else{
            printf("NO
");
        }
    }
    return 0;
}

分析：

1、自定义栈

2、通过两个栈模拟进出

方法二：

#include<stdio.h>
#include<iostream>
#include<vector>
#include<stack>
using namespace std;

int M,N,K;

int Check(vector<int> &v) {
    int i=0;
    int num=1;
    int cap=M+1;
    stack<int> sta;
    sta.push(0);
    while(i<N) {
        while(v[i]>sta.top()&&sta.size()<cap)
            sta.push(num++);
        if(v[i++]==sta.top())
            sta.pop();
        else
            return 0;
    }
    return 1;
}

int main() {
    vector<int> vec(N,0);
    scanf("%d%d%d",&M,&N,&K);
    for(int i=0; i<K; i++) {
        for(int j=0; j<N; j++) {
            int number;
            scanf("%d",&number);
            vec.push_back(number);
        }
        if(Check(vec)) printf("YES
");
        else printf("NO
");
        vec.clear();
    }
    return 0;
}

分析：

1、利用c++容器vector和stack模拟