小明正在做一个网络实验。
他设置了 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≤1e5 ,
1≤t≤100
分析:
并查集边带权
w记录增量,res记录这个节点的值
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e4 + 10;
int n,m,op,a,b,fa[maxn];
int w[maxn],res[maxn];
void Init(){
for(int i = 0; i < maxn; i++)
fa[i] = i;
}
int find(int x){
return fa[x] == x ? x : fa[x] = find(fa[x]);
}
void merge(int x,int y){
x = find(x);y=find(y);
if(x!=y){
for(register int i = 1; i <= n; i++){
res[i] += w[find(i)];
}
memset(w,0,sizeof(w));
fa[x] = y;
}
return ;
}
int main(){
ios::sync_with_stdio(0);
cin >> n >> m;
Init();
for(int i = 1; i <= m; i++){
cin >> op >> a >> b;
if(op==1){
merge(a,b);
}else{
w[find(a)] += b;
}
}
for(int i = 1; i <= n; i++){
if(i!=1) cout << " ";
cout << res[i] + w[find(i)];
}
return 0;
}