80分求助，第一个测试点过不了

#include<stdio.h>

int prime(int a)

{

    for (int i=2;i<a;i++)
    {
      if(a%i==0||a==1)
      
         return 1;
         
         }
         
    return 0;
    
}

int main()
{
	int n=0;
    
	int arr[1000];
    
	int e=0;
    
	  scanf("%d\n",&n);
      
	  for(int i=0;i<n;i++)
      
	  scanf("%d ",&arr[i]);
      
	  for(int j=0;j<n;j++){
      
	  	if(arr[j]==2)
        
	  	  printf("%d ",2);
          
	  	e=prime(arr[j]);
        
	  	    if(e==0)
            
	  	        printf("%d ",arr[j]);
                
	  }
	
	return 0;
}