結果

問題 No.2724 Coprime Game 1
コンテスト
ユーザー rlangevin
提出日時 2024-04-12 21:53:07
言語 C++17(clang)
(17.0.6 + boost 1.87.0)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,730 bytes
コンパイル時間 1,252 ms
コンパイル使用メモリ 124,288 KB
実行使用メモリ 222,336 KB
最終ジャッジ日時 2024-10-02 23:14:34
合計ジャッジ時間 19,509 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 3 TLE * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>

using namespace std;

class UnionFind {
private:
    vector<int> par;
    vector<int> rank;
    vector<int> size;

public:
    UnionFind(int n) {
        par.resize(n);
        rank.resize(n);
        size.resize(n);
        for (int i = 0; i < n; ++i) {
            par[i] = i;
            rank[i] = 0;
            size[i] = 1;
        }
    }

    int find(int x) {
        if (par[x] == x) {
            return x;
        } else {
            return par[x] = find(par[x]);
        }
    }

    void unionSet(int x, int y) {
        x = find(x);
        y = find(y);
        if (x != y) {
            if (rank[x] < rank[y]) {
                swap(x, y);
            }
            if (rank[x] == rank[y]) {
                rank[x]++;
            }
            par[y] = x;
            size[x] += size[y];
        }
    }

    bool isSame(int x, int y) {
        return find(x) == find(y);
    }

    int getSize(int x) {
        return size[find(x)];
    }
};

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);

    int T;
    cin >> T;

    const int M = 3 * 1000000 + 5;
    vector<int> L(M, 0);
    vector<vector<int>> G(M);
    for (int i = 2; i < M; ++i) {
        if (L[i]) {
            continue;
        }
        for (int j = 2 * i; j < M; j += i) {
            G[j].push_back(i);
            L[j] = 1;
        }
    }

    UnionFind uf(M);
    vector<int> ans(M, 0);
    for (int i = 2; i < M; ++i) {
        for (int j : G[i]) {
            uf.unionSet(i, j);
        }
        ans[i] = uf.getSize(i);
    }

    for (int _ = 0; _ < T; ++_) {
        int num;
        cin >> num;
        cout << (ans[num] % 2 == 0 ? "K" : "P") << endl;
    }

    return 0;
}
0