Description
Input
Output
Sample Input
23 8 51-143-12313 8 31-143-1231
Sample Output
YESNO | ||
#include<iostream>
#include<string>
using namespace std;
int main() {
int T,M,N,X,num,floor=0,sum;
int stack[1001];
cin>>T;
for(int i=0; i<T; i++) {
cin>>M;
cin>>N;
cin>>X;
floor=0;
sum=0;
int k=0;
for(; k<N; k++) {
cin>>num;
if(num==-1) {
floor--;
} else {
stack[floor]=num;
floor++;
}
}
floor--;
for(;stack[floor]!=M;floor--)
sum+=floor;
sum+=floor;
if(sum<=X)
cout<<"YES"<<endl;
else
cout<<"NO"<<endl;
}
}