题目要求
代码
#include <iostream>
#include <stack>
using namespace std;
int main(void){
const int N = 1005;
int arr[N]={0};
int m,n,k;
cin>>m>>n>>k;
for(int j=0;j<k;++j){ // k line
for(int i=1;i<=n;++i){
cin>>arr[i];
}
stack<int> st;
while(!st.empty()){ // empty stack
st.pop();
}
bool flag = true;
int num = 1;
for(int i=1;i<=n;++i){
st.push(i); // order
if(st.size()>m){// the maximum capacity
flag = false;
break;
}
while(st.empty()==false && st.top()==arr[num]){
st.pop();
++num;
}
}
if(flag && st.empty()){
cout<<"YES"<<endl;
}else{
cout<<"NO"<<endl;
}
}
return 0;
}
复习感悟
这次自己写感觉自己的方法比之前的简单,其实就是当i比测试数组中的大的时候就该pop了,小或等于的时候push,i为原始数据1~N,j是测试数组的下标
注意点:
1.每次对新的测试数据进行检测之前要先将堆栈清空
2.跳出循环break前,要将flag的改变置于break前
#include<iostream>
#include<stack>
using namespace std;
int main() {
int M, N, K;
cin >> M >> N >> K;
stack<int> s;
while (K--) {
int* test = new int[N];
for (int i = 0; i < N; i++)
cin >> test[i];
int i = 1, j = 0;
int flag = 0;
while (s.size() != 0)
s.pop();//保证堆栈中是空的
while (i <N+1||j<N) {
if (i > test[j]) {
int temp = s.top();
if (temp == test[j])
s.pop();
else { flag = 1; break; }
j++;
}
else if (i <= test[j]) {
if (s.size() < M)
{
s.push(i);
i++;
}
else { flag = 1; break; }//break要在cout后面
}
}
if (flag == 0)
cout << "YES" << endl;
else cout << "NO" << endl;
}
return 0;
}