线段树代码

# 线段树
# 对于满二叉树来说,当前节点下标i, 左儿子下标2i+1, 右儿子下标2i+2
#             0
#         1       2
#       3   4   5   6

def build_tree(arr, tree, node, start, end):  # node为根节点下标
    if start == end:
        tree[node] = arr[start]
        return

    mid = start + (end - start) // 2
    left_node = 2 * node + 1
    right_node = 2 * node + 2
    build_tree(arr, tree, left_node, start, mid)
    build_tree(arr, tree, right_node, mid + 1, end)
    tree[node] = tree[left_node] + tree[right_node]


def update_tree(arr, tree, node, start, end, idx, value):  # 更新arr[idx]=value
    if start == end:
        arr[idx] = value
        tree[node] = value
        return

    mid = start + (end - start) // 2
    left_node = 2 * node + 1
    right_node = 2 * node + 2
    if mid >= idx 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,我来帮你编写 Java 的线段代码线段是一种形结构,其中每个节点都存储了一段区间的信息。它常常用于处理区间查询、修改等问题。 下面是一个线段的简单实现,它可以处理区间求和问题: ``` class SegmentTree { private static class Node { int left; int right; int sum; } private Node[] tree; private int[] data; public SegmentTree(int[] data) { this.data = data; tree = new Node[data.length * 4]; build(0, 0, data.length - 1); } private void build(int treeIndex, int left, int right) { if (left == right) { tree[treeIndex] = new Node(); tree[treeIndex].left = left; tree[treeIndex].right = right; tree[treeIndex].sum = data[left]; return; } int mid = left + (right - left) / 2; int leftTreeIndex = treeIndex * 2 + 1; int rightTreeIndex = treeIndex * 2 + 2; build(leftTreeIndex, left, mid); build(rightTreeIndex, mid + 1, right); tree[treeIndex] = new Node(); tree[treeIndex].left = left; tree[treeIndex].right = right; tree[treeIndex].sum = tree[leftTreeIndex].sum + tree[rightTreeIndex].sum; } public void update(int index, int value) { update(0, index, value); } private void update(int treeIndex, int index, int value) { if (tree[treeIndex].left == tree[treeIndex].right) { tree[treeIndex].sum = value; return; } int mid = tree[treeIndex].left + (tree[treeIndex].right - tree[treeIndex].left) / 2; int leftTreeIndex = treeIndex * 2 + 1; int rightTreeIndex = treeIndex * 2 + 2; if (index <= mid) { update(leftTreeIndex, index, value); } else { update(rightTreeIndex, index, value); } tree[treeIndex].sum = tree[leftTreeIndex].sum + tree[rightTreeIndex].sum; } public int query(int left, int right) { return query(0, left, right); } private int query(int treeIndex, int left, int right) { if (left == tree[

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值