牛客网 集合合并 并查集路径压缩模板

题目描述
给定若干个32位int数字集合，每个集合中的数字无重复，譬如：
{1,2,3} {2,5,6} {8}
将其中交集不为空的集合合并，保证合并完成后所有集合之间无交集，输出合并后的集合个数以及最大集合中元素的个数。
输入描述:

输入格式：

1. 第一行为一个数字N，表示集合数。
2. 接下来N行，每行一个非空集合，包含若干个数字，数字之间用空格分开。
   假设第i个集合的大小为Si，数据满足N<=100000,ΣSi<=500000

输出描述:

输出格式：

1. 第一行为合并后的集合个数。
2. 第二个为最大集合中元素的个数。

示例1
输入
复制

3
1 2 3
2 5 6
8

输出
复制

2
5

#define debug
#ifdef debug
#include <time.h>
#include "/home/majiao/mb.h"
#endif

#include <iostream>
#include <algorithm>
#include <vector>
#include <string.h>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <math.h>

#define MAXN (500005)
#define ll long long int
#define INF (0x7f7f7f7f)
#define fori(lef, rig) for(int i=lef; i<=rig; i++)
#define forj(lef, rig) for(int j=lef; j<=rig; j++)
#define fork(lef, rig) for(int k=lef; k<=rig; k++)
#define QAQ (0)

using namespace std;

#define show(x...) \ do { \ cout << "\033[31;1m " << #x << " -> "; \ err(x); \ } while (0)

void err() { cout << "\033[39;0m" << endl; }
template<typename T, typename... A>
void err(T a, A... x) { cout << a << ' '; err(x...); }

namespace FastIO{

	char print_f[105];
	void read() {}
	void print() { putchar('\n'); }

	template <typename T, typename... T2>
		inline void read(T &x, T2 &... oth) {
			x = 0;
			char ch = getchar();
			ll f = 1;
			while (!isdigit(ch)) {
				if (ch == '-') f *= -1; 
				ch = getchar();
			}
			while (isdigit(ch)) {
				x = x * 10 + ch - 48;
				ch = getchar();
			}
			x *= f;
			read(oth...);
		}
	template <typename T, typename... T2>
		inline void print(T x, T2... oth) {
			ll p3=-1;
			if(x<0) putchar('-'), x=-x;
			do{
				print_f[++p3] = x%10 + 48;
			} while(x/=10);
			while(p3>=0) putchar(print_f[p3--]);
			putchar(' ');
			print(oth...);
		}
} // namespace FastIO
using FastIO::print;
using FastIO::read;

int n, m, Q, K;

string mul(string A, string B) {
	string ret(A.length() + B.length()+1, 0);
	for(auto& ch : A) ch -= '0';
	for(auto& ch : B) ch -= '0';
	reverse(A.begin(), A.end());
	reverse(B.begin(), B.end());
	int pos, flag; //flag进位标志
	for(int i=0; i<(int)A.length(); i++) {
		pos = i, flag = 0;
		for(int j=0; j<(int)B.length(); j++, pos++) {
			int t = (A[i] * B[j]) + flag + ret[pos];
			flag = t / 10;
			ret[pos] = t % 10;
		}
		ret[pos] += flag;
	}
	while(ret.back() == 0) ret.pop_back();
	for(auto& ch : ret) ch += '0';
	reverse(ret.begin(), ret.end());
	return ret;
}

int pre[MAXN], sum[MAXN];
int fa(int x) {
	return pre[x]==x ? x : pre[x]=fa(pre[x]);
}

void union_xy(int x, int y) {
	int rx = fa(x), ry = fa(y);
	if(rx != ry) {
		sum[rx] += sum[ry];
		pre[ry] = rx;
	}
}

int tot;
map<int, int> mp;
int ID(int x) {
	int& rx = mp[x];
	if(!rx) rx = ++ tot;
	return rx;
}

int main() {
#ifdef debug
	freopen("test", "r", stdin);
	clock_t stime = clock();
#endif
	scanf("%d ", &n);
	for(int i=1; i<MAXN; i++) pre[i] = i, sum[i] = 1;
	for(int i=1; i<=n; i++) {
		char ch;
		int x, lst = 500005;
		while(scanf("%d%c", &x, &ch)) {
			x = ID(x);
			if(lst != 500005) {
				union_xy(lst, x);
			}
			lst = x;
			if(ch == '\n') break;
		}
	}
	int cnt = 0, ans = 0;
	for(int i=1; i<=tot; i++) 
		if(pre[i] == i) {
			ans = max(ans, sum[i]);
			cnt ++;
		}
	printf("%d\n%d\n", cnt, ans);





#ifdef debug
	clock_t etime = clock();
	printf("rum time: %lf 秒\n",(double) (etime-stime)/CLOCKS_PER_SEC);
#endif 
	return 0;
}