๐Ÿ“ฆ ved015 / Codeforces-solution

๐Ÿ“„ newsdistribution1400.cpp ยท 89 lines
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89#include<iostream>
#include<vector>
#include<algorithm>
#include<map>
#include<set>
#include<string>
using namespace std;
typedef long long ll;
class DisjointSet {
public:
 
	vector<ll> rank, parent, size;
 
	DisjointSet(ll n) {
		rank.resize(n + 1, 0);
		parent.resize(n + 1);
		size.resize(n + 1, 1);
		for(ll i = 1; i <= n; i++) parent[i] = i;
	}
 
	void unionByRank(ll x, ll y) {
 
		ll par_x = findPar(x);
		ll par_y = findPar(y);
 
		if(par_x == par_y) return;
 
		if(rank[par_x] < rank[par_y]) {
			parent[par_x] = par_y;
		} else if(rank[par_y] < rank[par_x]) {
			parent[par_y] = par_x;
		} else {
			parent[par_x] = par_y;
			rank[par_y]++;
		}
	}
 
	void unionBySize(ll x, ll y) {
 
		ll par_x = findPar(x);
		ll par_y = findPar(y);
 
		if(par_x == par_y) return;
 
		if(size[par_x] < size[par_y]) {
			parent[par_x] = par_y;
			size[par_y] += size[par_x];
		} else {
			parent[par_y] = par_x;
			size[par_x] += size[par_y];
		}
	}
 
	ll findPar(ll x) {
 
		if(parent[x] == x) return x;
 
		return parent[x] = findPar(parent[x]);
 
	}
 
};
int main(){
    ll n, m;
    cin >> n >> m;
    
    DisjointSet ds(n);
    for(ll i = 0; i < m; i++) {
        ll x;
        cin >> x;
    
        if(x == 0) continue;
    
        ll u;
        cin >> u;
    
        for(ll j = 0; j < x - 1; j++) {
            ll v;
            cin >> v;
            ds.unionBySize(u, v);
        }
    }
    
    for(ll i = 1; i <= n; i++) {
        cout << ds.size[ds.findPar(i)] << " ";
    }
    cout << endl;
    return 0;
}