python解决 1051 Pop Sequence

M,N,K=[int(i) for i in input().split()]
for _ in range(K):
    A=[int(i) for i in input().split()]
    F,B,add=True,[],1
    for i in A:
        if len(B)==0 or B[-1]<i:   #当栈为空或者数不够大时，进行push操作
            while  add<=i:
                B.append(add)
                if len(B)>M:       #溢出时判错
                    F = False
                    break
                add+=1

        if i == B[-1]:            #相等时出站
            B.pop()
        elif i < B[-1]:           #小于时直接判错，因为堆栈的数是从大到小排列的
            print("NO")
            F = False
            break


        if F==False:
            print("NO")
            break
    if F:
        print("YES")

解题思路：

用list来模拟整个过程的压入弹出，每次读入一个数字的时候，说明这个数字之前的数字都已经被压入过栈中了，要不是在栈中，要不就是压入栈后已经被弹出了。声明一个空栈，这个空栈从压入1开始，依次递增，直到压入的数和读取的第一个数相等，说明这个时候需要对栈进行pop操作，依次对读取的第二，第三。。。第n个数进行模拟。在模拟的过程中要判断，这个栈中的数据是否已经超过了栈的容量m如果已经超过了m那个就没有继续进行模拟的必要了。