Subtask #1都AC了，Subtask #2为什么红红紫紫？求大佬解答。（会关注）

#include<bits/stdc++.h>
using namespace std;
long long n,s[10001]={},b;
int main(){
    cin>>n;string a;
	for(int i=1;i<=n;i++){
		cin>>a;
		if(a=="WA"||a=="TLE"||a=="MLE"){
			s[i]=1;
		}
	}
	for(int i=1;i<=n;i++){
    	if(s[i]==1)cout<<i<<" ";
    }
	return 0;
}