每日一题 第九十期 洛谷 [蓝桥杯 2020 省 AB1] 网络分析

[蓝桥杯 2020 省 AB1] 网络分析

题目描述

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

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

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

小明可以通过网线将两个节点连接起来，连接后两个节点就可以互相通信了。两个节点如果存在网线连接，称为相邻。

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

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

输入格式

输入的第一行包含两个整数 $n$，$m$，分别表示节点数量和操作数量。节点从 $1$ 至 $n$ 编号。
 
接下来 $m$ 行，每行三个整数，表示一个操作。

如果操作为 1 a b，表示将节点 $a$ 和节点 $b$ 通过网线连接起来。当 $a=b$ 时，表示连接了一个自环，对网络没有实质影响。

如果操作为 2 p t，表示在节点 $p$ 上发送一条大小为 $t$ 的信息。

输出格式

输出一行，包含 $n$ 个整数，相邻整数之间用一个空格分割，依次表示进行完上述操作后节点 $1$ 至节点 $n$ 上存储信息的大小。

样例 #1

样例输入 #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

提示

对于 $30\%$ 的评测用例，$1\le n\le 20$，$1\le m\le 100$。

对于 $50\%$ 的评测用例，$1\le n\le 100$，$1\le m\le 1000$。

对于 $70\%$ 的评测用例，$1\le n\le 1000$，$1\le m\le 10000$。

对于所有评测用例，$1\le n\le 10000$，$1\le m\le 10^5$ ，$1\le t\le 100$。

蓝桥杯 2020 第一轮省赛 A 组 J 题（B 组 J 题）。

AC代码：

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef pair<int, int>PII;
const int N=3e5+10;
const int MOD=9901;
const int INF=0X3F3F3F3F;
const int dx[]={-1,1,0,0,-1,-1,+1,+1};
const int dy[]={0,0,-1,1,-1,+1,-1,+1};
const int M = 1e4 + 10;

int n, m;
int p[M];
int d[M];
int val[M];
int finds(int x)
{
	if(x == p[x] || p[p[x]] == p[x]) return p[x];
	int r = finds(p[x]);
	d[x] += d[p[x]];
	p[x] = r;
	return r;
}
int main()
{
	cin >> n >> m;
	for(int i = 1; i <= n; i ++) p[i] = i;
	while(m --){
		int l, a, b;
		cin >> l >> a >> b;
		if(l == 1){
			a = finds(a);
			b = finds(b);
			if(a != b)
			{
				d[a] -= d[b];
				p[a] = b;
			}
		}
	    else {
			a = finds(a);
			d[a] += b;
		}
	}
	for(int i = 1; i <= n; i ++)
	{
		if(i == finds(i)) cout << d[i] << " ";
		else cout << d[i] + d[finds(i)] << " ";
	}
	cout << endl;
	return 0;
}