POJ 3468 A Simple Problem with Integers

区间修改查询树

最新推荐文章于 2023-10-02 10:26:46 发布

原创最新推荐文章于 2023-10-02 10:26:46 发布 · 524 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#线段树

POJ 专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种基于线段树的数据结构实现，支持区间修改和区间查询操作，适用于解决需要批量更新及快速查询的问题。文中提供了完整的C++代码实现，包括初始化、更新节点以及查询等核心功能。

区间修改+区间查询

#include<iostream>
using namespace std;
const int M=100005;
struct Seb_tree
{
int l;
int r;
long long col;
long long sum;
};
Seb_tree t[M*4+1];
void push_down(int k)
{
if (t[k].col)
{
int l=k*2;
int r=l+1;
t[l].sum+=t[k].col*(t[l].r-t[l].l+1);
t[l].col+=t[k].col;
t[r].sum+=t[k].col*(t[r].r-t[r].l+1);
t[r].col+=t[k].col;
t[k].col=0;
}
}
long long query(int l, int r, int k)
{
if (t[k].l==l && t[k].r==r)
{
return t[k].sum;
}
push_down(k);
int mid=(t[k].r+t[k].l)/2;
if (r<=mid) query(l ,r, k*2);
else if (l>mid) query(l, r, k*2+1);
else return query(l, mid, k*2)+query(mid+1, r, k*2+1);
}
void updata(int l, int r, int col, int k)
{
if (t[k].l>=l && t[k].r<=r)
{
t[k].sum+=col*(r-l+1);
t[k].col+=col;
return;
}
push_down(k);
int mid=(t[k].r+t[k].l)/2;
if (r<=mid) updata(l ,r, col, k*2);
else if (l>mid) updata(l, r, col, k*2+1);
else {
updata(l, mid, col, k*2);
updata(mid+1, r, col, k*2+1);
}
t[k].sum=t[k*2].sum+t[k*2+1].sum;
}
void build(int l, int r, int k)
{
t[k].l=l;
t[k].r=r;
t[k].col=0;
if (l==r){
long long s;
cin>>s;
t[k].sum=s;
return;
}
int mid=(l+r)/2;
build(l, mid, k*2);
build(mid+1, r, k*2+1);
t[k].sum=t[k*2] .sum+t[k*2+1].sum;
}
int main()
{
int n,m;
cin>>n>>m;
build(1, n, 1);
while (m--)
{
char a;
int b, c, d;
cin>>a;
if (a=='C'){
cin>>b>>c>>d;
updata(b, c, d, 1);
}
else if (a=='Q')
{
cin>>b>>c;
cout<<query(b,c,1)<<endl;
}

}
return 0;
}