洛谷P3374模板线段树1

本文介绍了一种高效的数据结构——段式树,并结合懒惰传播技术来解决数列更新与查询问题。通过实例演示了如何使用段式树进行区间加法更新及区间和查询操作。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目描述

如题,已知一个数列,你需要进行下面两种操作:

1.将某一个数加上x

2.求出某区间每一个数的和

输入输出格式

输入格式:
第一行包含两个整数N、M,分别表示该数列数字的个数和操作的总个数。

第二行包含N个用空格分隔的整数,其中第i个数字表示数列第i项的初始值。

接下来M行每行包含3或4个整数,表示一个操作,具体如下:

操作1: 格式:1 x k 含义:将第x个数加上k

操作2: 格式:2 x y 含义:输出区间[x,y]内每个数的和

输出格式:
输出包含若干行整数,即为所有操作2的结果。

输入输出样例

输入样例#1:
5 5
1 5 4 2 3
1 1 3
2 2 5
1 3 -1
1 4 2
2 1 4
输出样例#1:
14
16
说明

时空限制:1000ms,128M

数据规模:

对于30%的数据:N<=8,M<=10

对于70%的数据:N<=10000,M<=10000

对于100%的数据:N<=500000,M<=500000

样例说明:

这里写图片描述

故输出结果14、16
板子要多撸几遍。。。

#include <iostream>

using namespace std;

#define int long long

struct node
{
    int l,r,sum;
}tree[300001];
int a[100001],f[100001];
int lazy[300001];
void buildTree(int x,int l,int r)
{
    tree[x].l=l;
    tree[x].r=r;
    if(l==r)
    {
        tree[x].sum=a[l];
        return;
    }
    buildTree(x*2,l,(l+r)>>1);
    buildTree(x*2+1,1+((l+r)>>1),r);
    tree[x].sum=tree[x*2].sum+tree[x*2+1].sum;
}
void pushdown(int x)
{
    int mid=(tree[x].l+tree[x].r)/2;
    tree[x*2].sum+=(mid-tree[x].l+1)*lazy[x];
    tree[x*2+1].sum+=(tree[x].r-mid)*lazy[x];
    lazy[x*2]+=lazy[x];
    lazy[x*2+1]+=lazy[x];
    lazy[x]=0;
}
void update(int x,int l,int r,int k)
{
    if(tree[x].l>=l&&tree[x].r<=r)
    {
        tree[x].sum+=(tree[x].r-tree[x].l+1)*k;
        lazy[x]+=k;
        return;
    }
    if(tree[x].l>r||tree[x].r<l)
        return;
    int mid=(tree[x].r+tree[x].l)/2;
    if(lazy[x])pushdown(x);
    update(x*2,l,r,k);
    update(x*2+1,l,r,k);
    tree[x].sum=tree[x*2].sum+tree[x*2+1].sum;
}
int query(int x,int l,int r)
{
    if(tree[x].l>=l&&tree[x].r<=r)
        return tree[x].sum;
    if(tree[x].l>r||tree[x].r<l)
        return 0;
    if(lazy[x])pushdown(x);
    return query(x*2,l,r)+query(x*2+1,l,r);
}
signed main()
{
    int n,m,t,x,y,z;
    cin >> n >> m;
    for(int i=1;i<=n;i++)
        cin >> a[i];
    buildTree(1,1,n);
    for(int i=1;i<=m;i++)
    {
        cin>>t>>x>>y;
        if(t==1){cin>>z;update(1,x,y,z);}
        else cout<<query(1,x,y)<<endl;
    }
    return 0;
}
### Python 实现的线段树模板代码 对于洛谷平台上的题目 `P3372 【模板线段树 1`,可以采用如下所示的 Python 版本的线段树实现方法[^1]。 ```python class SegmentTree: def __init__(self, data): self.n = len(data) self.tree = [0] * (4 * self.n) # 初始化线段树数组大小为原数据长度四倍 self.build(1, 0, self.n - 1, data) def build(self, node, start, end, data): if start == end: self.tree[node] = data[start] else: mid = (start + end) >> 1 self.build(node << 1, start, mid, data) self.build((node << 1) + 1, mid + 1, end, data) self.push_up(node) def update(self, idx, val): self._update(1, 0, self.n - 1, idx, val) def _update(self, node, start, end, idx, val): if start == end: self.tree[node] = val else: mid = (start + end) >> 1 if start <= idx <= mid: self._update(node << 1, start, mid, idx, val) else: self._update((node << 1) + 1, mid + 1, end, idx, val) self.push_up(node) def query(self, L, R): return self._query(1, 0, self.n - 1, L, R) def _query(self, node, start, end, L, R): if R < start or end < L: return 0 # 返回一个不影响最终结果的值 elif L <= start and end <= R: return self.tree[node] mid = (start + end) >> 1 sum_left = self._query(node << 1, start, mid, L, R) sum_right = self._query((node << 1) + 1, mid + 1, end, L, R) return sum_left + sum_right def push_up(self, node): self.tree[node] = self.tree[node << 1] + self.tree[(node << 1) + 1] if __name__ == "__main__": n, m = map(int, input().split()) a = list(map(int, input().split())) seg_tree = SegmentTree(a) results = [] for _ in range(m): cmd, x, y = input().split() x = int(x) y = int(y) if cmd == 'Q': result = seg_tree.query(x-1, y-1) results.append(result) elif cmd == 'U': seg_tree.update(x-1, y) for res in results: print(res) ``` 此代码实现了基本的线段树功能,包括构建、更新以及查询操作。特别地,在初始化阶段创建了一个足够大的列表来存储所有可能被访问到的位置,并在线程中递归地建立子区间的表示形式。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值