网络分析 并查集的应用 蓝桥杯 java

小明正在做一个网络实验。

他设置了 n 台电脑，称为节点，用于收发和存储数据。

初始时，所有节点都是独立的，不存在任何连接。

小明可以通过网线将两个节点连接起来，连接后两个节点就可以互相通信了。

两个节点如果存在网线连接，称为相邻。

小明有时会测试当时的网络，他会在某个节点发送一条信息，信息会发送到每个相邻的节点，之后这些节点又会转发到自己相邻的节点，直到所有直接或间接相邻的节点都收到了信息。

所有发送和接收的节点都会将信息存储下来。

一条信息只存储一次。

给出小明连接和测试的过程，请计算出每个节点存储信息的大小。

输入格式
输入的第一行包含两个整数 n,m，分别表示节点数量和操作数量。

节点从 1 至 n 编号。

接下来 m 行，每行三个整数，表示一个操作。

如果操作为 1 a b，表示将节点 a 和节点 b 通过网线连接起来。当 a = b 时，表示连接了一个自环，对网络没有实质影响。
如果操作为 2 p t，表示在节点 p 上发送一条大小为 t 的信息。
输出格式
输出一行，包含 n 个整数，相邻整数之间用一个空格分割，依次表示进行完上述操作后节点 1 至节点 n 上存储信息的大小。

数据范围
1≤n≤10000,
1≤m≤105,
1≤t≤100
输入样例1：
4 8
1 1 2
2 1 10
2 3 5
1 4 1
2 2 2
1 1 2
1 2 4
2 2 1
输出样例1：
13 13 5 3

import java.util.*;

public class Main
{
	static int n,m,k;
	static int N=20005;
	static int f[]=new int[N],d[]=new int[N];
	
	static int find(int x)
	{
		if(x==f[x]||f[x]==f[f[x]])return f[x];
		int r=find(f[x]);//执行完这条，原来的父节点已经更新，直接连在根节点
		d[x]+=d[f[x]];
		f[x]=r;
		return r;
	}
	
	public static void main(String args[])
	{
		Scanner sc=new Scanner(System.in);
		n=sc.nextInt();
		m=sc.nextInt();
		k=n+1;
		for(int i=1;i<N;++i)f[i]=i;
		while((m--)>0)
		{
			int t,a,b;
			t=sc.nextInt();
			a=sc.nextInt();
			b=sc.nextInt();
			if(t==1)
			{
				a=find(a);
				b=find(b);
				if(a!=b)
				{
					f[a]=k;
					f[b]=k++;
				}
			}
			else
			{
				a=find(a);
				d[a]+=b;
			}
		}
		for(int i=1;i<=n;++i)
		{
			if(i==find(i))System.out.print(d[i]+" ");
			else System.out.print(d[i]+d[f[i]]+" ");
		}
	}
}