两篇讲解线段树的文章 链接1 链接2
目录
POJ - 3468
简单线段树。
#include<iostream>
#include<stdio.h>
#include<cstring>
#include<string>
using namespace std;
const int maxn=100010;
int arr[maxn];
long long ans=0;
struct node
{
int l,r;
long long w;
long long add;
};
node t[maxn*4];
long long up(int k)
{
t[k].w=t[2*k].w+t[2*k+1].w;
return t[k].w;
}
void down(int k)
{
if(t[k].add!=0)
{
t[2*k].add+=t[k].add;
t[2*k+1].add+=t[k].add;
t[2*k].w+=(t[2*k].r-t[2*k].l+1)*t[k].add;
t[2*k+1].w+=(t[2*k+1].r-t[2*k+1].l+1)*t[k].add;
t[k].add=0;
}
}
void build(int l,int r,int k)
{
t[k].l=l;t[k].r=r;t[k].add=0;
if(l==r)
{
return;
}
int mid=(l+r)/2;
build(l,mid,2*k);
build(mid+1,r,2*k+1);
up(k);
}
void update(int l,int r,int add,int k)
{
if(t[k].l>=l&&t[k].r<=r)
{
t[k].w+=(t[k].r-t[k].l+1)*add;
t[k].add+=add;
return;
}
down(k);
int mid=(t[k].l+t[k].r)/2;
if(l<=mid) update(l,r,add,2*k);
if(r>mid) update(l,r,add,2*k+1);
up(k);
}
void query(int l,int r,int k)
{
if(t[k].l>=l&&t[k].r<=r)
{
ans+=t[k].w;
return;
}
down(k);
int mid=(t[k].l+t[k].r)/2;
if(l<=mid) query