POJ 1363

题意：判断一个栈是否能输出相应的数据

 

思路：

从头开始检查要求的输出。要求数组为a的话，判断a[i]这一位是否可能输出。显然a[i]这一位如果能够按照要求输出，它一定是未压入栈的或者是在栈顶。如果未压入栈，则把它压入栈，当然挡住它的也都被压进去了。如果在栈顶，弹出顶部元素。如果都不符合，输出No

 

统计：200k, 63ms, 2Y，一次把Yes和No的大小写弄错了

 

 #include <iostream>#include <stack>#include <string>#define F(i,a,b) for (int i=a;i<=b;i++)using namespace std;stack<int> st;int a[1001], Q[1001];bool instack[1001], wait[1001];int main(){string line;int n;while (scanf("%d", &n) && n!=0){while (1){ bool done = false; F(i,1,n) { scanf("%d", &a[i]); if (a[i] == 0) { done = true; break; } } if (done) break;//solvewhile (!st.empty()) st.pop();F(i,1,n)Q[i] = i;memset(wait, true, sizeof(wait));memset(instack, 0, sizeof(instack) );int head = 1;int i;for (i=1;i<=n;i++){if ( wait[ a[i] ]){while ( Q[head] != a[i] ){st.push( Q[head] );wait[ Q[head] ] = false;instack[ Q[head] ] = true;head++;}wait[ a[i] ]=false;head++;}elseif (st.top() == a[i]){st.pop();instack[a[i]]=false;}else{printf("No/n");break;}}if (i==n+1) printf("Yes/n"); }printf("/n");}return 0;}