蓝桥杯ACwing习题

题意 : 两个操作 :

  1. 给定 a b a 和 b 连一条边

  2. 给 a 上加一个权值所有和 a 直接或间接相连的边都加上这个权值

第一个操作很简单就是维护并查集的基本操作 当 a != b 是 p[a] = b

第一个操作会将所有的点变成一棵树的形式 然后当给一个点上加一个值的时候 就相当于这个树的根节点

加了一个权值 在后面要求每一个点的权值时候 如果这个点不是根节点的话 肯定是当前点的权值加上其所属的连通块根节点的权值 d[i] + d[find(i)]

注意 :在合并树的时候如果直接合并的话会出现 当前连通块内部所有联通块的点 都会加上合并后的那个连通块根节点的值,所以要用 d[a] - d[b]

还有就是路径压缩的时候常用的就是求当前点到根节点的距离

那就是 d[x] += d[p[x]] 在每一次回溯的时候当前的点都会加上父节点的权值 这样就可以保证每一个在当前连通块里面的点都会加上根节点的值

代码

#include <iostream>

#include <cstring>

#include <algorithm>

using namespace std;

const int N = 1e5 + 10;

int n , m;

int p[N];

int d[N];

int find(int x)

{

if(!(x == p[x] || p[p[x]] == p[x]))

{

int t = find(p[x]);

d[x] += d[p[x]];

p[x] = t;

}

return p[x];

}

int main()

{

cin >> n >> m;

for(int i = 1 ; i <= n ; i ++) p[i] = i;

for(int i = 0 ; i < m ; i ++)

{

int op , a , b;

cin >> op >> a >> b;

if(op == 1)

{

a = find(a) , b = find(b);

if(a != b)

{

d[a] -= d[b];

p[a] = b;

}

}

else

{

a = find(a);

d[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;

}

相关推荐
DuHz7 小时前
论文精读:大语言模型 (Large Language Models, LLM) —— 一项调查
论文阅读·人工智能·深度学习·算法·机器学习·计算机视觉·语言模型
加农炮手Jinx7 小时前
LeetCode 72. Edit Distance 题解
算法·leetcode·力扣
借雨醉东风7 小时前
程序分享--常见算法/编程面试题:旋转矩阵
c++·线性代数·算法·面试·职场和发展·矩阵
_深海凉_7 小时前
LeetCode热题100-打家劫舍
算法·leetcode·职场和发展
jghhh018 小时前
使用 MATLAB 实现支持向量回归 (SVR) 预测未来数据
算法·matlab
云泽8088 小时前
笔试算法 - 双指针篇(二):四大经典求和题型 + 有效三角形计数问题
c++·算法
刀法如飞9 小时前
【合并已排序数组的三种实现策略,哪一种更可取?】
算法·程序员
王老师青少年编程9 小时前
csp信奥赛C++高频考点专项训练之贪心算法 --【区间贪心】:种树
c++·算法·贪心·csp·信奥赛·区间贪心·种树
hi_ro_a10 小时前
C++ 哈希表封装 unordered_map /unordered_set
数据结构·c++·算法·哈希算法
Jasmine_llq13 小时前
《B4447 [GESP202512 二级] 环保能量球》
数据结构·算法·数学公式计算(核心)·整数除法算法·多组数据循环处理·输入输出算法·简单模拟算法