牛客oj 習題11.2 第一題(並查集+set)

 

題目鏈接：click here

思路：挺抽象的一道題，他沒說有多少個節點，只給出道路以及其相連的點，那所有的節點就是已知道路對應的節點總和。這裏用set存儲最合適，因爲要去重。剩下就是標準的並查集了。另外這題沒給出數據範圍，那麼久老規矩100w。

 

 

 

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <string>
#include <cstring>
#include <vector>
#include <queue>
#include <set>
#include <climits>
 
using namespace std;
 
const int MAXN = 1000005;
const int INF = INT_MAX;

int father[MAXN];
int height[MAXN];

int Find(int x){
	if(x != father[x]) father[x] = Find(father[x]);
	return father[x];
}

void Union(int x, int y){
	x = Find(x);
	y = Find(y);
	if(x != y){
		if(height[x] < height[y]) father[x] = y;
		else if(height[y] < height[x]) father[y] = x;
		else{
			father[y] = x;
			height[x]++;
		}
	}
}

void Initial(){
	for(int i = 0; i < MAXN; i++){
		father[i] = i;
		height[i] = 0;
	}
}

int main(){
//	freopen("in.txt", "r", stdin);
	int i, j;
	Initial();
	set<int> myset;
	while(~scanf("%d %d", &i, &j)){
		Union(i, j);
		myset.insert(i);
		myset.insert(j);
	}
	int ans = 0;
	set<int>::iterator it; 
	for(it = myset.begin(); it != myset.end(); it++){
		int current = *it;
		if(current == father[current]) ans++;
	}
	printf("%d\n", ans);
	return 0;
}