結果

問題 No.826 連絡網
コンテスト
ユーザー misora192
提出日時 2020-07-23 22:52:18
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 41 ms / 2,000 ms
コード長 1,588 bytes
コンパイル時間 1,666 ms
コンパイル使用メモリ 173,836 KB
実行使用メモリ 15,160 KB
最終ジャッジ日時 2024-06-23 21:48:44
合計ジャッジ時間 3,051 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define rep(i,n) for(int i=(0);i<(n);i++)

using namespace std;

typedef long long ll;

template<class T> bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; }
template<class T> bool chmin(T &a, const T &b) { if (a>b) { a=b; return 1; } return 0; }

// union-find tree with size(struct)
struct UnionFind{
    vector<int> par; // 親ノード
    vector<int> rank; // ランク
	vector<int> size; // 連結成分のサイズ

    UnionFind(int n = 1) {
        init(n);
    }

    void init(int n = 1) {
        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;

        int r = find(par[x]);
        return par[x] = r;
    }

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

    bool unite(int x, int y) {
	    x = find(x);
		y = find(y);
	    if (x == y) return false;

	    if (rank[x] < rank[y]) swap(x, y);
	    if (rank[x] == rank[y]) ++rank[x];
	    par[y] = x;
		size[x] += size[y];
	    return true;
    }

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

	int getRank(int x){
		return rank[find(x)];
	}
};


int main(){
	cin.tie(0);
	ios::sync_with_stdio(false);

	int n, p;
	cin >> n >> p;

	UnionFind uf(n+1);
	vector<bool> isprime(n+1, true);
	isprime[0] = isprime[1] = false;
	for(int i = 2; i <= n; i++){
		if(isprime[i]){
			for(int j = 2 * i; j <= n; j += i){
				uf.unite(i, j);
				isprime[j] = false;
			}
		}
	}

	cout << uf.getSize(p) << endl;
}
0