样例正确，但是0分，求条

#include <iostream>
using namespace std;
int is_prime(int a) {
    if (a == 0 or a == 1) {
        return 0;
    } else {
        for (int i = 2; i <= a; i++) {
            if (a % i == 0) {
                return 0;
            }
            return 1;
        }
    }
}

int main() {
    int n;
    cin >> n;
    int a[101];
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
    }
    for (int i = 1; i <= n; i++) {
        if(is_prime(a[i])) {
            cout << a[i] << " ";
        }
    }
    return 0;
}