pta ds c語言 7-2 出棧序列的合法性 (25分) 簡單思路

7-2 出棧序列的合法性 (25分)

 

給定一個最大容量為 M 的堆棧，將 N 個數字按 1, 2, 3, ..., N 的順序入棧，允許按任何順序出棧，則哪些數字序列是不可能得到的？例如給定 M=5、N=7，則我們有可能得到{ 1, 2, 3, 4, 5, 6, 7 }，但不可能得到{ 3, 2, 1, 7, 5, 6, 4 }。

輸入格式：

輸入第一行給出 3 個不超過 1000 的正整數：M（堆棧最大容量）、N（入棧元素個數）、K（待檢查的出棧序列個數）。最后 K 行，每行給出 N 個數字的出棧序列。所有同行數字以空格間隔。

輸出格式：

對每一行出棧序列，如果其的確是有可能得到的合法序列，就在一行中輸出YES，否則輸出NO。

輸入樣例：

5 7 5
1 2 3 4 5 6 7
3 2 1 7 5 6 4
7 6 5 4 3 2 1
5 6 4 3 7 2 1
1 7 6 5 4 3 2

 

輸出樣例：

YES
NO
NO
YES
NO

 

作者

陳越

單位

浙江大學

代碼長度限制

16 KB

時間限制

400 ms

內存限制

 

 

順序入棧 1 2 3 4 5 6 7 

對於任意 5 6 4 3 7 2 1

關注其合法性 是這樣的一個步驟

對於任意一個元素比如 6 其后（指右邊）元素比6小的 必須是降序的 如 4 3 2 1

 原因很簡單 因為順序入棧 比6小的元素要么在棧外 在6的左邊，要么在棧中就是它的右邊

 

因為題目給了棧的最大容量  所以我又先檢驗了這個棧的長度是否合理 這兩部我分開做 合在一起會快很多 期待讀者實現

 

#include<stdio.h>
#include<stdlib.h>

int main(){
    int m,n,k;
    scanf("%d%d%d",&m,&n,&k);
    int a[k][n];
    int legal[k];
    for(int i=0;i<k;i++)legal[i]=1;
    for(int i=0;i<k;i++){
        for(int j=0;j<n;j++){
            scanf("%d",&a[i][j]);
        }
    }
    //對出棧合法性檢驗包括兩個部分 
    //1.棧長度檢驗 不超過m  
    for(int i=0;i<k;i++){
        for(int j=0;j<n-m;j++){
            int cnt = 1;
            for(int z = j+1 ; z < n; z++){
                 if(a[i][z]<a[i][j])cnt++;
            }
            if(cnt>m)
            {
                 legal[i]=0;
                 break;
            }
        }
    }
    // 2.還在棧中的較小的元素 必須是遞減的
    for(int i=0;i<k;i++){
        for(int j=0;j<n;j++){
            int premin = a[i][j];
            for(int z=j+1;z<n;z++){
                if(a[i][z] < a[i][j]){
                    if(a[i][z]>premin){
                        legal[i] = 0;
                        break;
                    }
                    else{
                        premin = a[i][z];
                    }
                }
            }
        }
    } 
    for(int i=0;i<k;i++){
        if(legal[i])printf("YES\n");
        else printf("NO\n");
    } 
    return 0;
}