蓝桥C++省赛B组--J:网络分析

网络分析

题目如下

试题 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站也有并查集的视频讲解。