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

P8710 [蓝桥杯 2020 省 AB1] 网络分析 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

题意：

        将每台电脑连接起来，并求出其储存的信息。操作1是将a和b连接起来，操作2是给连接p的电脑发送大小为t的信息

思路：

        连接操作我们可以联想到并查集的操作，将a和b连接起来，但是发送信息的这个过程，如果暴力找一遍的话，时间复杂度是O(n²)会tle。

        那么我们需要换一个思路，在连接的时候将存储的信息加到ans里，并将祖先定为层数小的那个，加信息的时候存储到另一个数组里放起来，不急着加上去先。那么这样会有一个问题，最后一步可能不是连接两个点，那么最后还要加上存储数组里对应的值。

代码：

#include<bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;
const int N = 1e4+10;
int w[N],fa[N],val[N],dep[N];
int find(int x)
{
	if (fa[x] != x) fa[x] = find(fa[x]);
	return fa[x];
}
signed main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	
	int n,m,op,p,t;
	cin >> n >> m;
	for (int i = 1; i <= n; ++i) fa[i] = i;
	while(m--)
	{
		cin >> op >> p >> t;
		if (op == 1){
			if (find(p) == find(t)) continue;
			for (int i = 1; i <= n; ++i) w[i] += val[find(i)];
			for (int i = 1; i <= n; ++i) val[i] = 0;
			p = find(p) , t = find(t); // 合并
			if (dep[p] > dep[t]) swap(p,t);
			fa[p] = t; // 层数小的当祖先
			dep[t] += dep[p];
		}
		if (op == 2) val[find(p)] += t;
	}
	for (int i = 1; i <= n; ++i) cout << w[i] + val[find(i)] << " ";
	return 0;
}