pat 1051

最新推荐文章于 2024-04-25 10:28:26 发布

原创最新推荐文章于 2024-04-25 10:28:26 发布 · 333 阅读

0 ·

CC 4.0 BY-SA版权

Pat 专栏收录该内容

101 篇文章

订阅专栏

本文介绍了一个用于验证特定序列是否可以通过栈操作实现的算法。该算法通过标记数组跟踪元素的状态，并利用标准模板库中的栈来模拟操作流程。适用于容量受限的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

a[i] = 0 未进stack
a[i] = 1 已在stack
a[i] = 2 已出stack
正确的两种情况
1.若s.top == sequence[i] pop之
2.若s.top != sequence[i] 将小于sequence[i]的且未在stack内，也未出stack的push进去，并且s.size < capacity

#include<stdio.h>
#include<stack>
using namespace std;
bool CheckStack(int sequence[], int n, int capacity){
    int *a = new int[n + 1];
    int i,j;
    for (i = 1; i <= n; i++)
        a[i] = 0;
    stack<int> s;
    int flag;
    for (i = 0; i < n; i++){
        flag = 0;
        if (s.size() == 0){
            for (j = 1; j <= sequence[i]; j++)
                if (!a[j]){
                    a[j] = 1;
                    s.push(j);
                }
            if (s.size() > capacity)
                break;
        }
        if (s.top() == sequence[i]){
                s.pop();
                a[sequence[i]] = 2;
                flag = 1;
        }
        else if (s.top() != sequence[i] && a[sequence[i]] == 0){
                for (j = 1; j <= sequence[i]; j++)
                    if (!a[j]){
                        a[j] = 1;
                        s.push(j);
                }
                if (s.size() > capacity)
                    break;
                s.pop();
                a[sequence[i]] = 2;
                flag = 1;
        }
        if (!flag)
            break;
    }
    if (i == n)
        return true;
    else return false;
}
int main(){
    freopen("1.in", "r", stdin);
    int n, times, capacity;
    scanf("%d%d%d", &capacity, &n, &times);
    int *a = new int[n];
    int i;
    while (times--){
        for (i = 0; i < n; i++)
            scanf("%d", &a[i]);
        bool flag = CheckStack(a, n, capacity);
        if (flag)
            printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}