C语言经典算法之线段树查询

JJJ69

于 2024-01-29 08:26:00 发布

阅读量1k

点赞数 26

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

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

版权

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

160 篇文章 12 订阅

订阅专栏

本文详细介绍了线段树的数据结构，包括C语言的代码实现，讨论了其时间复杂度（O(logn)查询，O(n)建树）和空间复杂度（O(n)）。同时，分析了线段树的优缺点，并列举了其在现实世界中的各种应用，如区间和查询、动态数组操作等。

摘要由CSDN通过智能技术生成

前言

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;
}

// 查询区间和
int querySegmentTree(SegmentTreeNode* root, int start, int end) {
    // 如果查询区间超出当前节点的范围，返回0
    if (root == NULL || start > root->end || end < root->start) {
        return 0;
    }

    // 如果当前节点完全包含在查询区间内，返回当前节点的和
    if (start <= root->start && end >= root->end) {
        return root->sum;
    }

    // 否则，递归查询左右子树
    int leftSum = querySegmentTree(root->left, start, end);
    int rightSum = querySegmentTree(root->right, start, end);

    // 返回左右子树的和
    return leftSum + rightSum;
}

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);

    return 0;
}

这个简单的例子演示了如何用C语言实现线段树的构建和查询操作。你可以根据具体的问题需求对这个示例进行修改和扩展。