小明正在做一个网络实验。
他设置了 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
并查集做法
#include<bits/stdc++.h>
using namespace std;
const int N=100100;
int p[N],d[N];
int n,m,t;
int find(int x)
{
if (p[x] != x)
{
if (p[p[x]] == p[x]) return p[x];
d[x] += d[p[x]]; // 因为要进行路径压缩 ,所以这里要加上他的父节点的权值;
int r = find(p[x]);
p[x] = r;
}
return p[x];
}
int main(){
cin >> n >> m;
int k=n;
for(int i=0;i<N;i++) p[i]=i; // 因为后面要增加新节点 ,所以这里要遍历到N;
while(m--){
int a,b;
cin >> t >> a >> b;
if(t==1){
if(find(a)!=find(b)){
k++;
p[find(a)]=p[find(b)]=k;//指向新节点
}
}
else {
d[find(a)]+=b;
}
}
for (int i = 1; i <= n; i ++ )
if (i == find(i)) printf("%d ", d[i]);
else printf("%d ", d[i] + d[find(i)]);
return 0;
}
这里图中给出的做法是将一个根节点指向另一个根节点,上述代码中是建立一个新的根节点,大体思路还是相同的
暴力做法 会超时
还就内个暴搜
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
vector<int>m[N];
bool st[N];
int a[N];
int n,t;
void dfs(int now,int father,int score){
for(int i=0;i<m[now].size();i++){
if(m[now][i]!=father&&!st[m[now][i]]){
a[m[now][i]]+=score;
st[m[now][i]]=true;
dfs(m[now][i],now,score);
}
}
}
int main(){
cin >> n >> t;
while(t--){
int op;
scanf("%d",&op);
if(op==1){
int c,d;
scanf("%d%d",&c,&d);
int flag=0;
for(int i=0;i<m[c].size();i++){
if(d==m[c][i]) flag=1;
}
if(!flag&&c!=d){
m[c].push_back(d);
m[d].push_back(c);
}
}
if(op==2){
memset(st,false,sizeof st);
int c,d;
scanf("%d%d",&c,&d);
a[c]+=d;
st[c]=true;
dfs(c,-1,d);
}
}
for(int i=1;i<=n;i++){
cout << a[i] << " ";
}
return 0;
}