AcWing 2069.网格分析(并查集)
作者:
冰冷酒
,
2022-02-26 18:33:50
,
所有人可见
,
阅读 154
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
const int N = 200010;
int p[N], d[N], n, m;
void init() // 数组初始化
{
for (int i = 0; i < N; i ++)
p[i] = i;
}
int find(int x)
{
if (x != p[x])
{
if (p[x] == p[p[x]]) // 如果他的父节点是祖宗结点或自己是祖宗结点,直接返回
return p[x];
int t = find(p[x]);
d[x] += d[p[x]]; // 在路径压缩过程中更新点权以及父节点
p[x] = t;
}
return p[x];
}
int main()
{
ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
init();
cin >> n >> m;
while (m --)
{
int w;
cin >> w;
if (w == 1)
{
int a, b;
cin >> a >> b;
int fa = find(a), fb = find(b);
if (fa != fb)
{
p[fa] = fb;
d[fa] -= d[fb]; // 两颗树相连的时候为保证点权正确,要做减法(手推一下即可)、也可以重新定义一个共同的父节点(权值为0)
}
}
else
{
int p, t;
cin >> p >> t;
int fp = find(p);
d[fp] += t;
}
}
for (int i = 1; i <= n; i ++)
{
if (i == find(i)) cout << d[i] << ' '; // 如果自己为根节点,那直接输出自己就好
else cout << d[i] + d[find(i)] << ' '; // 如果自己不是根节点,就要输出自己的权值+祖宗结点的权值
}
cout << '\n';
return 0;
}