C语言经典算法之线段树更新

JJJ69

于 2024-01-29 08:44:01 发布

阅读量799

点赞数 31

分类专栏： C语言经典算法文章标签：算法 c语言数据结构开发语言

本文链接：https://blog.csdn.net/weixin_56154577/article/details/135904095

版权

C语言经典算法专栏收录该内容

160 篇文章 13 订阅

订阅专栏

前言

A.建议

1.学习算法最重要的是理解算法的每一步，而不是记住算法。

2.建议读者学习算法的时候，自己手动一步一步地运行算法。

tips：文中的（如果有）对数，则均以2为底数

B.简介

线段树更新算法主要用于在给定的线段树中更新某个特定节点的值，以及通过递归更新其父节点的值。

一代码实现

假设我们要解决的问题是在一个数组中查询某个区间的和，并支持对数组元素的更新。首先，我们需要建立线段树，然后可以通过更新算法来修改树中相应节点的值。

#include <stdio.h>
#include <stdlib.h>

// 线段树节点的定义
typedef struct {
    int start, end;  // 区间的起始和结束位置
    int sum;         // 区间的和
} SegmentTreeNode;

// 构建线段树
SegmentTreeNode* buildSegmentTree(int* arr, int start, int end) {
    if (start > end) {
        return NULL;
    }

    // 创建当前节点
    SegmentTreeNode* node = (SegmentTreeNode*)malloc(sizeof(SegmentTreeNode));
    node->start = start;
    node->end = end;

    // 如果是叶子节点，直接将数组中的值赋给节点
    if (start == end) {
        node->sum = arr[start];
    } else {
        // 如果不是叶子节点，递归构建左右子树，并计算当前节点的和
        int mid = (start + end) / 2;
        node->sum = 0; // 初始化为0
        node->left = buildSegmentTree(arr, start, mid);
        node->right = buildSegmentTree(arr, mid + 1, end);

        // 计算当前节点的和
        if (node->left != NULL) {
            node->sum += node->left->sum;
        }
        if (node->right != NULL) {
            node->sum += node->right->sum;
        }
    }

    return node;
}

// 更新线段树节点
void updateSegmentTree(SegmentTreeNode* root, int index, int newValue) {
    // 检查是否超出当前节点的范围
    if (root == NULL || index < root->start || index > root->end) {
        return;
    }

    // 如果是叶子节点，更新节点的值
    if (root->start == root->end) {
        root->sum = newValue;
        return;
    }

    // 否则，递归更新左右子树
    updateSegmentTree(root->left, index, newValue);
    updateSegmentTree(root->right, index, newValue);

    // 更新当前节点的和
    root->sum = root->left->sum + root->right->sum;
}

// 查询线段树
int querySegmentTree(SegmentTreeNode* root, int start, int end) {
    // 查询逻辑，与之前介绍的查询算法相似，不再重复给出
    // ...
}

int main() {
    int arr[] = {1, 3, 5, 7, 9, 11};
    int n = sizeof(arr) / sizeof(arr[0]);

    // 构建线段树
    SegmentTreeNode* root = buildSegmentTree(arr, 0, n - 1);

    // 查询区间和
    int queryStart = 1, queryEnd = 4;
    int result = querySegmentTree(root, queryStart, queryEnd);
    printf("Sum of elements in the range [%d, %d] is %d\n", queryStart, queryEnd, result);

    // 更新线段树节点
    int updateIndex = 2, newValue = 8;
    updateSegmentTree(root, updateIndex, newValue);

    // 重新查询区间和
    result = querySegmentTree(root, queryStart, queryEnd);
    printf("Updated sum of elements in the range [%d, %d] is %d\n", queryStart, queryEnd, result);

    return 0;
}

这个示例中，updateSegmentTree 函数用于更新线段树中的某个节点的值。在主函数中，我们通过构建线段树、查询区间和、更新节点值等操作展示了线段树的基本使用和更新算法的实现。