PTA天梯賽L2-032 彩虹瓶（c語言實作）

原題連結

這道題說實話就是題目有點繞，但其實的意思就是堆棧的一個變式應用。

最核心的代碼我寫了注釋，感興趣的朋友可以交流一下。

#include <stdio.h>
typedef struct{
	int stack[1001];
	int top;
}stack;
void init(stack *s){
	s->top=0;
} 
void push(stack *s,int a){
	s->stack[s->top++]=a;
}
void pop(stack *s){
	s->top--;
}
int gettop(stack s){
	if(s.top!=0){
		return s.stack[s.top-1];
	}
}
int main(){
	int n,m,k,i,j,l,t;
	scanf("%d %d %d",&n,&m,&k);
	for(l=0;l<k;l++){
		int p=1;
		stack s;
		init(&s);
		int x[n],y[n];
		for(i=0;i<n;i++){
			y[i]=0;
			scanf("%d",&x[i]);
		}
		i=0;
		t=1;//設t的初值為1，表示從1号顔色的小球開始裝
		while(t<=n){//當小球編号小于等于n的時候執行這個循環
		//當目前指向的數組中的值不等于t 棧頂的值不等于t 并且t已經入棧過了
		//這時勞工會發怒
			if(x[i]!=t && gettop(s)!=t && y[t]==1){
				p=0;
				break;
			}//當目前指向的數組中的值等于t，i和t都加1
			if(x[i]==t){
				i++;
				t++;
				continue;
			}//當棧頂的值等于t，退棧一次，t加1
			if(gettop(s)==t){
				t++;
				pop(&s);
				continue;
			}//當棧頂指針指到m時說明堆棧中已經有m個數了，不能繼續入棧
			if(s.top==m){
				p=0;
				break;
			}
			push(&s,x[i]);//如果都不是上述情況，入棧，i+1
			i++;
		}
		if(p){
			printf("YES\n");
		}else{
			printf("NO\n");
		}
	}
}