哪里有问题

#include<stdio.h>
#include<math.h>
int zhi(int n)
{
    if(n==1) return 0;
    if(n==2) return 1;
    if(n%2==0) return 0;
    for(int i=3;i<=n-1;i++)
    {
        if(n%i==0) return 0;
        else return 1;
    }

}
int main()
{
    int n,a;
    scanf("%d",&n);
    int arr[n];
    for(int i=1;i<=n;i++)
    {
        scanf("%d ",&a);
        arr[i]=a;
        if(zhi(arr[i]))
        {
            printf("%d ",arr[i]);
        }
    }
    return 0;
}