网络分析
题目如下
试题 J: 网络分析
问题描述
时间限制: 1.0s 内存限制: 256.0MB 本题总分:25 分
【问题描述】
小明正在做一个网络实验。
他设置了 n 台电脑,称为节点,用于收发和存储数据。
初始时,所有节点都是独立的,不存在任何连接。
小明可以通过网线将两个节点连接起来,连接后两个节点就可以互相了。两个节点如果存在网线连接,称为相邻。
小明有时会测试当时的网络,他会在某个节点发送一条信息,信息会到每个相邻的节点,之后这些节点又会转发到自己相邻的节点,直到所有或间接相邻的节点都收到了信息。所有发送和接收的节点都会将信息存储一条信息只存储一次。
给出小明连接和测试的过程,请计算出每个节点存储信息的大小。
【输入格式】
输入的第一行包含两个整数 n, m,分别表示节点数量和操作数量。节1 至 n 编号。
接下来 m 行,每行三个整数,表示一个操作。
如果操作为 1 a b,表示将节点 a 和节点 b 通过网线连接起来。当 时,表示连接了一个自环,对网络没有实质影响。
如果操作为 2 p t,表示在节点 p 上发送一条大小为 t 的信息。
【输出格式】
输出一行,包含 n 个整数,相邻整数之间用一个空格分割,依次表示完上述操作后节点 1 至节点 n 上存储信息的大小。
【样例输入】
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
2
3
4
5
6
7
8
9
【样例输出】
13 13 5 3
1
【评测用例规模与约定】
对于 30% 的评测用例,1 ≤ n ≤ 20,1 ≤ m ≤ 100。
对于 50% 的评测用例,1 ≤ n ≤ 100,1 ≤ m ≤ 1000。
对于 70% 的评测用例,1 ≤ n ≤ 1000,1 ≤ m ≤ 10000。
对于所有评测用例,1 ≤ n ≤ 10000,1 ≤ m ≤ 100000,1 ≤ t ≤ 100。
对于这题我们可以用并差集来写
代码如下:
#include <iostream>
#include <vector>
using namespace std;
class union_find{
public:
int count; //联通分量的数量
vector<int> parent; //合并关系
vector<int> size; //树的weight
vector<int> info; //每个节点的消息量
public:
union_find(int capacity); //初始化
void Union(int p,int q){ //连接节点
int rootp = parent[p];
int rootq = parent[q];
if(rootp == rootq){
return ;
}
if(size[rootp] > size[rootq]){ //用weight防止树失衡退化
parent[rootq] = rootp;
size[rootp] += size[rootq];
}else{
parent[rootp] = rootq;
size[rootq] += size[rootp];
}
--count;
//cout << "union successful!" << endl;
}
int find(int x){ //查找节点的父节点
while(parent[x] != x){
//进行路径压缩
parent[x] = parent[parent[x]];
x = parent[x];
}
return x;
}
/*
int conn_cnt(){ //联通分量个数,在这题没用
return count;
}
*/
bool connected(int p,int q){ //判断两个节点是否联结起来了
int rootp = find(p);
int rootq = find(q);
return rootp == rootq;
}
void send(int p,int t){ //题目中说的发送消息
for(int i = 0;i < info.size();++i){
if(connected(p,i)){
info[i] += t;
}
}
//cout << "send successful!" << endl;
}
void ans(){ //输出结果
for(int x : info){
cout << x << " ";
}
}
};
union_find::union_find(int capacity) //初始化
{
count = capacity;
parent.resize(capacity);
size.resize(capacity);
info.resize(capacity);
for(int i = 0;i < count;++i){
parent[i] = i; //父节点指向自己
size[i] = 1; //初始大小都是1
info[i] = 0; /初始都没有接收到任何信息
}
//cout << "初始化成功!" << endl;
}
int main(){
int num = 0,oper =0;
cin >> num >> oper;
union_find uf(num);
for(int i = 0;i < oper;++i){
int flag = 0,a,b;
cin >> flag >> a >> b;
if(flag == 1){ //根据操作进行判断
uf.Union(a - 1,b - 1);
}else{
uf.send(a - 1,b);
}
}
uf.ans();
return 0;
}
大部分细节都写在注释里,如果不了解并查集可以去搜索相关资料,b站也有并查集的视频讲解。