写代码c语言已知有n个节点,有n-1条边,形成一个树的结构. 给定一个根节点S.每个节点都有一个权值,节点i的权值为vi. 给出m个操作.操作有两种类型: 1 a x ,表示将节点a的权值加上x 2 a ,表示求a节点的子树上所有节点的和(包括a节点本身)

时间: 2024-04-26 11:20:27 浏览: 5
下面是使用C语言实现的代码,基于树的遍历实现: ```c #include <stdio.h> #include <stdlib.h> #define MAXN 100005 typedef long long ll; int n, m; // 节点数和操作数 int v[MAXN]; // 节点权值 int head[MAXN], nxt[MAXN * 2], to[MAXN * 2], cnt = 0; // 邻接表存储图 int fa[MAXN], sz[MAXN], son[MAXN], top[MAXN], dep[MAXN], id[MAXN], rk[MAXN], tot = 0; // 树剖相关数组 ll sum[MAXN << 2], add[MAXN << 2]; // 线段树相关数组 inline void add_edge(int u, int v) { nxt[++cnt] = head[u]; head[u] = cnt; to[cnt] = v; } void dfs1(int u, int f) { fa[u] = f; sz[u] = 1; dep[u] = dep[f] + 1; for (int i = head[u]; i; i = nxt[i]) { int v = to[i]; if (v == f) continue; dfs1(v, u); sz[u] += sz[v]; if (sz[v] > sz[son[u]]) son[u] = v; } } void dfs2(int u, int t) { top[u] = t; id[u] = ++tot; rk[tot] = u; if (!son[u]) return; dfs2(son[u], t); for (int i = head[u]; i; i = nxt[i]) { int v = to[i]; if (v == fa[u] || v == son[u]) continue; dfs2(v, v); } } inline void pushup(int i) { sum[i] = sum[i << 1] + sum[i << 1 | 1]; } inline void pushdown(int i, int l, int r) { if (add[i] != 0) { int mid = (l + r) >> 1; add[i << 1] += add[i]; sum[i << 1] += (mid - l + 1) * add[i]; add[i << 1 | 1] += add[i]; sum[i << 1 | 1] += (r - mid) * add[i]; add[i] = 0; } } void build(int i, int l, int r) { if (l == r) { sum[i] = v[rk[l]]; return; } int mid = (l + r) >> 1; build(i << 1, l, mid); build(i << 1 | 1, mid + 1, r); pushup(i); } void update(int i, int l, int r, int ql, int qr, int x) { if (ql <= l && r <= qr) { add[i] += x; sum[i] += (r - l + 1) * x; return; } pushdown(i, l, r); int mid = (l + r) >> 1; if (ql <= mid) update(i << 1, l, mid, ql, qr, x); if (qr > mid) update(i << 1 | 1, mid + 1, r, ql, qr, x); pushup(i); } ll query(int i, int l, int r, int ql, int qr) { if (ql <= l && r <= qr) return sum[i]; pushdown(i, l, r); int mid = (l + r) >> 1; ll ans = 0; if (ql <= mid) ans += query(i << 1, l, mid, ql, qr); if (qr > mid) ans += query(i << 1 | 1, mid + 1, r, ql, qr); return ans; } void modify(int x, int y) { update(1, 1, n, id[x], id[x], y); } ll query_subtree(int x) { ll ans = 0; while (top[x] != top[1]) { ans += query(1, 1, n, id[top[x]], id[x]); x = fa[top[x]]; } ans += query(1, 1, n, 1, id[x]); return ans; } int main() { scanf("%d %d", &n, &m); for (int i = 1; i <= n; i++) scanf("%d", &v[i]); for (int i = 1; i < n; i++) { int u, v; scanf("%d %d", &u, &v); add_edge(u, v); add_edge(v, u); } dfs1(1, 0); dfs2(1, 1); build(1, 1, n); while (m--) { int op, a, x; scanf("%d %d", &op, &a); if (op == 1) { scanf("%d", &x); modify(a, x); } else { printf("%lld\n", query_subtree(a)); } } return 0; } ``` 代码中使用了树剖和线段树两种数据结构,其中树剖用于快速求子树和,线段树用于维护节点权值的修改。在每次操作时,如果是修改操作,则直接在对应节点上加上给定的值,如果是查询操作,则使用树剖求出对应节点的子树范围(从该节点到其子树的最深叶子节点),然后在线段树上查询该范围内所有节点权值之和即可。

相关推荐

最新推荐

recommend-type

高分项目 基于STM32F103单片机的无线测距系统源代码+项目资料齐全+教程文档.zip

【资源概览】 高分项目 基于STM32F103单片机的无线测距系统源代码+项目资料齐全+教程文档.zip高分项目 基于STM32F103单片机的无线测距系统源代码+项目资料齐全+教程文档.zip高分项目 基于STM32F103单片机的无线测距系统源代码+项目资料齐全+教程文档.zip 【资源说明】 高分项目源码:此资源是在校高分项目的完整源代码,经过导师的悉心指导与认可,答辩评审得分高达95分,项目的质量与深度有保障。 测试运行成功:所有的项目代码在上传前都经过了严格的测试,确保在功能上完全符合预期,您可以放心下载并使用。 适用人群广泛:该项目不仅适合计算机相关专业(如电子信息、物联网、通信工程、自动化等)的在校学生和老师,还可以作为毕业设计、课程设计、作业或项目初期立项的演示材料。对于希望进阶学习的小白来说,同样是一个极佳的学习资源。 代码灵活性高:如果您具备一定的编程基础,可以在此代码基础上进行个性化的修改,以实现更多功能。当然,直接用于毕业设计、课程设计或作业也是完全可行的。 欢迎下载,与我一起交流学习,共同进步!
recommend-type

java八股文介绍00002

java八股文,"Java八股文"通常指的是在面试Java开发者时经常被问到的一系列标准问题,这些问题往往是技术面试中的基础部分,用来评估应聘者对Java编程语言和Java虚拟机(JVM)的理解程度。这些问题的答案往往模式化,因此被称为“八股文”。虽然这个词汇带有一定的贬义,但是这些问题却是面试中不可或缺的一部分,因为它们能够快速地检验应聘者的基本技能和理论知识。
recommend-type

东北大学本科毕业设计 论文latex模板 2019

东北大学本科毕业设计 论文latex模板 2019 关于东北大学本科毕业设计论文的LaTeX模板,您可以参考以下两个资源: 1. **Acytoo/neu_bachelor_thesis_template**: 这个模板是修改自一位学长的模板,以适应东北大学最新的要求。它使用GBT7714-2005参考文献标准,并提供了详细的安装和使用说明。您需要使用xelatex进行编译,并可能需要手动安装参考文献的标准。此外,如果您使用的是VSCode编辑器,可以参考提供的VSCode配置。 2. **tzaiyang/NEUBachelorThesis**: 这个模板同样适用于东北大学本科生毕业论文。它提供了详细的VSCode配置,包括快捷键设置和texcount bug修复(仅限Windows用户)。此外,还提供了如何插入公式、图片和三线表的说明,以及如何引用参考文献的方法。 您可以根据自己的需求和喜好选择其中一个模板。如果您对LaTeX不熟悉,建议先学习一些基础知识,以便更好地使用这些模板。祝您毕业设计顺利!
recommend-type

node-v7.2.0-linux-armv7l.tar.xz

Node.js,简称Node,是一个开源且跨平台的JavaScript运行时环境,它允许在浏览器外运行JavaScript代码。Node.js于2009年由Ryan Dahl创立,旨在创建高性能的Web服务器和网络应用程序。它基于Google Chrome的V8 JavaScript引擎,可以在Windows、Linux、Unix、Mac OS X等操作系统上运行。 Node.js的特点之一是事件驱动和非阻塞I/O模型,这使得它非常适合处理大量并发连接,从而在构建实时应用程序如在线游戏、聊天应用以及实时通讯服务时表现卓越。此外,Node.js使用了模块化的架构,通过npm(Node package manager,Node包管理器),社区成员可以共享和复用代码,极大地促进了Node.js生态系统的发展和扩张。 Node.js不仅用于服务器端开发。随着技术的发展,它也被用于构建工具链、开发桌面应用程序、物联网设备等。Node.js能够处理文件系统、操作数据库、处理网络请求等,因此,开发者可以用JavaScript编写全栈应用程序,这一点大大提高了开发效率和便捷性。 在实践中,许多大型企业和组织已经采用Node.js作为其Web应用程序的开发平台,如Netflix、PayPal和Walmart等。它们利用Node.js提高了应用性能,简化了开发流程,并且能更快地响应市场需求。
recommend-type

串口通讯原理+硬件原理解释+软件配置

STM32串口通信大纲---适用于初学者 I. 引言 学习背景与串口通信的重要性 简述本文目的与主要内容 II. STM32里的串口通信 基本概念与功能介绍 USART及STM32内置串口 全双工通信与异步时钟控制 接收/发送逻辑结构 III. 串口配置 选择合适的串口资源(USART1/2/3) 参数设置:波特率、位数、校验、停止位 GPIO配置:TX/RX引脚驱动方式 IV. 串口发送与接收 发送数据流程 设置TDR与发送移位寄存器 检测TXE标志位 数据移位与持续发送 接收数据流程 配置RX输入模式 读取RDR与RXNE标志 字节数据接收 V. 实现方式 不使用中断的简单示例 利用中断接收数据 查询中断状态 中断服务程序处理数据
recommend-type

zigbee-cluster-library-specification

最新的zigbee-cluster-library-specification说明文档。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

实现实时数据湖架构:Kafka与Hive集成

![实现实时数据湖架构:Kafka与Hive集成](https://img-blog.csdnimg.cn/img_convert/10eb2e6972b3b6086286fc64c0b3ee41.jpeg) # 1. 实时数据湖架构概述** 实时数据湖是一种现代数据管理架构,它允许企业以低延迟的方式收集、存储和处理大量数据。与传统数据仓库不同,实时数据湖不依赖于预先定义的模式,而是采用灵活的架构,可以处理各种数据类型和格式。这种架构为企业提供了以下优势: - **实时洞察:**实时数据湖允许企业访问最新的数据,从而做出更明智的决策。 - **数据民主化:**实时数据湖使各种利益相关者都可
recommend-type

如何用python编写api接口

在Python中编写API接口可以使用多种框架,其中比较流行的有Flask和Django。这里以Flask框架为例,简单介绍如何编写API接口。 1. 安装Flask框架 使用pip命令安装Flask框架: ``` pip install flask ``` 2. 编写API接口 创建一个Python文件,例如app.py,编写以下代码: ```python from flask import Flask, jsonify app = Flask(__name__) @app.route('/api/hello', methods=['GET']) def hello():
recommend-type

JSBSim Reference Manual

JSBSim参考手册,其中包含JSBSim简介,JSBSim配置文件xml的编写语法,编程手册以及一些应用实例等。其中有部分内容还没有写完,估计有生之年很难看到完整版了,但是内容还是很有参考价值的。