线段树模板
区间修改区间查询
例题 codevs 1082线段树练习3(注意sum要开long long!)
#include <iostream>
#include <cstdio>
const int maxn = 200010;
int num[maxn];
struct node{
int l,r;
long long sum,delta;
}t[maxn << 2];
void build(int cur,int l,int r){
t[cur].l = l;
t[cur].r = r;
t[cur].delta = 0;
if (l+1 == r){
t[cur].sum = num[l];
return;
}
build(cur << 1,l,(l+r)/2);
build((cur << 1)+1,(l+r)/2,r);
t[cur].sum = t[cur << 1].sum + t[(cur << 1)+1].sum;
}
void update(int cur,int delta){
t[cur << 1].sum += delta*(t[cur << 1].r-t[cur << 1].l);
t[cur << 1].delta += delta;
t[(cur << 1)+1].sum += delta*(t[(cur << 1)+1].r-t[(cur << 1)+1].l);
t[(cur << 1)+1].delta += delta;
t[cur].delta = 0;
}
void change (int cur,int l,int r,int delta){
if (l<=t[cur].l && t[cur].r<=r){
t[cur].sum += delta*(t[cur].r-t[cur].l);
t[cur].delta += delta;
return;
}
int mid = (t[cur].l + t[cur].r)/2;
if (t[cur].delta != 0) update(cur,t[cur].delta);
if (l<mid) change(cur << 1,l,r,delta);
if (r>mid) change((cur << 1)+1,l,r,delta);
t[cur].sum = t[cur << 1].sum+t[(cur << 1)+1].sum;
}
long long query(int cur,int l,int r){
if (l<=t[cur].l && t[cur].r<=r) return t[cur].sum;
long long sum(0),mid = (t[cur].l + t[cur].r)/2;
if (t[cur].delta != 0) update(cur,t[cur].delta);
if (l<mid) sum += query(cur << 1,l,r);
if (r>mid) sum += query((cur << 1)+1,l,r);
return sum;
}
int main(){
int n,m,f,x,y,delta;
scanf("%d",&n);
for (int i=1;i<=n;i++)
scanf("%d",&num[i]);
build(1,1,n+1);
scanf("%d",&m);
for (int i=0;i<m;i++){
scanf("%d%d%d",&f,&x,&y);
if (f == 1){
scanf("%d",&delta);
change(1,x,y+1,delta);
}
else printf("%lld\n",query(1,x,y+1));
}
return 0;
}
修改最小值,判断是否小于0
例题 codevs 1217 借教室(NOIP2012提高组)
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 1000010;
int num[maxn];
struct node{
int l,r;
long long min,delta;
}t[maxn << 2];
void build(int cur,int l,int r){
t[cur].l = l;
t[cur].r = r;
t[cur].delta = 0;
if (l+1 == r){
t[cur].min = num[l];
return;
}
build(cur << 1,l,(l+r)/2);
build((cur << 1)+1,(l+r)/2,r);
t[cur].min = min(t[cur << 1].min,t[(cur << 1)+1].min);
}
void update(int cur,int delta){
t[cur << 1].min += delta;
t[cur << 1].delta += delta;
t[(cur << 1)+1].min += delta;
t[(cur << 1)+1].delta += delta;
t[cur].delta = 0;
}
bool change (int cur,int l,int r,int delta){
if (l<=t[cur].l && t[cur].r<=r){
t[cur].min += delta;
t[cur].delta += delta;
if (t[cur].min<0) return 0;
return 1;
}
int mid = (t[cur].l + t[cur].r)/2,ans(1);
if (t[cur].delta != 0) update(cur,t[cur].delta);
if (l<mid) ans &= change(cur << 1,l,r,delta);
if (r>mid) ans &= change((cur << 1)+1,l,r,delta);
t[cur].min = min(t[cur << 1].min,t[(cur << 1)+1].min);
return ans;
}
int main(){
int n,m,f,x,y,delta;
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++)
scanf("%d",&num[i]);
build(1,1,n+1);
for (int i=0;i<m;i++){
scanf("%d%d%d",&delta,&x,&y);
if (!change(1,x,y+1,-delta)){
printf("-1\n%d\n",i+1);
return 0;
}
}
printf("0\n");
return 0;
}