51Nod 1831 PN表

打出PN表来 发现合数除16,34,289都是赢 质数除2,17都是输

#include<bits/stdc++.h>
using namespace std;
bool prime(int x) {
        for (int i = 2; i <= sqrt(x); i++) {
                if (x % i == 0) {
                        return false;
                }
        }
        return true;
}
int main() {
        int n;
        scanf("%d", &n);
        while (n--) {
                int x;
                scanf("%d", &x);
                if (x == 1) {
                        printf("NIE\n");
                } else {
                        if (prime(x)) {
                                if (x == 2 || x == 17) {
                                        printf("TAK\n");
                                } else {
                                        printf("NIE\n");
                                }
                        } else {
                                if (x == 16 || x == 34 || x == 289) {
                                        printf("NIE\n");
                                } else {
                                        printf("TAK\n");
                                }
                        }
                }
        }
}