Codeforces - C . News Distribution

In some social network, there are n users communicating with each other in m groups of friends. Let’s analyze the process of distributing some news between users.

Initially, some user x receives the news from some source. Then he or she sends the news to his or her friends (two users are friends if there is at least one group such that both of them belong to this group). Friends continue sending the news to their friends, and so on. The process ends when there is no pair of friends such that one of them knows the news, and another one doesn’t know.

For each user x you have to determine what is the number of users that will know the news if initially only user x starts distributing it.

Input
The first line contains two integers n and m (1≤n,m≤5⋅105) — the number of users and the number of groups of friends, respectively.

Then m lines follow, each describing a group of friends. The i-th line begins with integer ki (0≤ki≤n) — the number of users in the i-th group. Then ki distinct integers follow, denoting the users belonging to the i-th group.

It is guaranteed that ∑i=1mki≤5⋅105.

Output
Print n integers. The i-th integer should be equal to the number of users that will know the news if user i starts distributing it.

Example
inputCopy
7 5
3 2 5 4
0
2 1 2
1 1
2 6 7
outputCopy
4 4 1 4 4 2 2

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<ctime>
#include<iostream>
#include<algorithm>
#include<map>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<string>
#define ll long long
#define dd double
using namespace std;

ll n, m;
ll arr[5000040];
ll number[5000040];

ll arr1(ll w) {
	if (arr[w] != w) {
		arr[w] = arr1(arr[w]);
		return arr[w];
	}
	else {
		return w;
	}
}

void arr2(ll q1, ll q2) {
	ll t1;
	ll t2;
	t1 = arr1(q1);
	t2 = arr1(q2);
	if (t1 != t2) {
		arr[t2] = t1;
		number[t1] += number[t2];
	}
	return;
}

int main() {
	ios::sync_with_stdio(false);
	cin >> n >> m;
		for (ll i = 0; i <= n; i++) {
			arr[i] = i;
			number[i] = 1;
		}
		while (m--) {
			ll a, b, c;
			cin >> a;
			if (a != 0) {
				cin >> b;
				for (ll i = 1; i < a; i++) {
					cin >> c;
					arr2(b, c);
					b = c;
				}
			}
		}
		for (ll q = 1; q <= n; q++) {
			cout << number[arr1(q)];
			if (q == n) {
				cout << endl;
			}
			else {
				cout << " ";
			}
		}
}