算法——线段树C/C++

算法梦想家

于 2024-09-03 20:17:44 发布

阅读量1.6k

点赞数 38

文章标签：算法 c++ 数据结构 c语言图论线段树

本文链接：https://blog.csdn.net/2401_86771711/article/details/141870725

版权

线段树（Segment Tree）是一种用于处理区间查询和区间更新的高效数据结构。它可以将一个数组分成多个小区间，并在每个小区间上存储一些信息（如区间的最小值、最大值、和等），以便快速查询或更新整个数组中的某个区间。下面，我将详细介绍如何使用C++实现一个基本的线段树，用于查询区间和以及更新单点值。

线段树的基本结构

线段树是一个二叉树，其中每个节点代表数组的一个区间。每个节点存储的信息通常是该区间内所有元素的一个汇总（例如，和、最小值、最大值等）。对于节点node[i]，其左子节点为node[2*i+1]，右子节点为node[2*i+2]（假设数组索引从0开始）。

1. 定义节点结构体

首先，我们需要定义线段树的节点结构体，通常包含区间的起始和结束位置，以及该区间的汇总信息。

#include <vector> 
#include <iostream> 


using namespace std; 


struct SegmentTreeNode { 
int start, end; 
long long sum; // 假设存储的是和 
SegmentTreeNode *left, *right; 


SegmentTreeNode(int s, int e) : start(s), end(e), sum(0), left(nullptr), right(nullptr) {} 
};

2. 构建线段树

接下来，我们需要一个函数来构建线段树。

SegmentTreeNode* buildTree(const vector<int>& nums, int start, int end) { 
if (start > end) return nullptr; 
SegmentTreeNode* node = new SegmentTreeNode(start, end); 
if (start == end) { 
node->sum = nums[start]; 
return node; 
} 
int mid = (start + end) / 2; 
node->left = buildTree(nums, start, mid); 
node->right = buildTree(nums, mid + 1, end); 
node->sum = node->left->sum + node->right->sum; 
return node; 
}

3. 单点更新

更新线段树中某个位置的值，并向上更新所有祖先节点的信息。

void update(SegmentTreeNode* node, int index, int value) { 
if (node->start == node->end) { 
node->sum = value; 
return; 
} 
int mid = (node->start + node->end) / 2; 
if (index <= mid) { 
update(node->left, index, value); 
} else { 
update(node->right, index, value); 
} 
node->sum = node->left->sum + node->right->sum; 
}

4. 区间查询

查询线段树中某个区间的和。

long long query(SegmentTreeNode* node, int start, int end) { 
if (start > node->end || end < node->start) return 0; 
if (start <= node->start && node->end <= end) { 
return node->sum; 
} 
int mid = (node->start + node->end) / 2; 
long long leftSum = query(node->left, start, end); 
long long rightSum = query(node->right, start, end); 
return leftSum + rightSum; 
}

5. 主函数和测试

int main() { 
vector<int> nums = {1, 3, 5, 7, 9, 11}; 
SegmentTreeNode* root = buildTree(nums, 0, nums.size() - 1); 


cout << "Initial sum of range [1, 3]: " << query(root, 1, 3) << endl; // 应输出 15 


update(root, 2, 4); // 将索引为2的元素从5更新为4 
cout << "Sum of range [1, 3] after update: " << query(root, 1, 3) << endl; // 应输出 14 


return 0; 
}