第一个测试点WA，其余全部AC，不知道哪里错了

代码如下：

#include<iostream>
#include<cmath>

using namespace std;
int a[10010];
void prime(int x,int y){
	bool flag=true;
	for(int i=2;i*i<=x;i++){
		if(x%i==0){
			flag=false;
			break;
		}
	}
	if(flag==false){
		a[y]=0;
	}
}
int main(){
	int n;
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>a[i];
		prime(a[i],i);
	}
	for(int i=0;i<n;i++){
		if(a[i]!=0){
			cout<<a[i]<<' ';
		}
	}
	return 0;
}