文章标题

最新推荐文章于 2023-10-19 15:03:50 发布

wosiguwozai0133

最新推荐文章于 2023-10-19 15:03:50 发布

阅读量317

点赞数

分类专栏：线段树

本文链接：https://blog.csdn.net/wosiguwozai0133/article/details/52346866

版权

线段树专栏收录该内容

4 篇文章 0 订阅

订阅专栏

本文介绍了如何使用平衡二叉树（如BST的变种）来高效地处理区间加权求和的问题。通过BuildTree、Insert和Quiry函数，实现了对区间加权值的插入和查询操作，支持动态更新和快速查询。代码中还包括了一个简单的输入读取函数readch。

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

include

using namespace std;

const int maxn = 100001;

struct Node
{
int L,R;
long long sum,inc;
};

Node Tree[maxn*3];

void BuildTree(int root,int L,int R)
{
Tree[root].L = L,Tree[root].R = R;
Tree[root].sum = 0,Tree[root].inc = 0;
if(L != R)
{
int mid = (Tree[root].L+Tree[root].R)/2;
BuildTree(2*root,L,mid);
BuildTree(2*root+1,mid+1,R);
}
}

void Insert(int root,int s,int e,int h)
{
if(Tree[root].L == s && Tree[root].R == e)
{
Tree[root].inc += h;
return;
}
Tree[root].sum += (e-s+1)*h;
int mid = (Tree[root].L+Tree[root].R)/2;
if(e <= mid)
Insert(2*root,s,e,h);
else if(s > mid)
Insert(2*root+1,s,e,h);
else
{
Insert(2*root,s,mid,h);
Insert(2*root+1,mid+1,e,h);
}
}

long long Quiry(int root,int l,int r)
{
if(Tree[root].L == l && Tree[root].R == r)
{
return (Tree[root].sum+(r-l+1)*Tree[root].inc);
}
long long inc_value = (r-l+1)*Tree[root].inc;
int mid = (Tree[root].L+Tree[root].R)/2;
if(r <= mid)
return (Quiry(2*root,l,r)+inc_value);
else if(l > mid)
return (Quiry(2*root+1,l,r)+inc_value);
else
{
return(Quiry(2*root,l,mid)+Quiry(2*root+1,mid+1,r)+inc_value);
}
}
char readch()
{
char ch;
while(ch = getchar(),(ch<’A’|| ch > ‘Z’ ));
return ch;
}