CSU-2151: 集训难度
Time Limit: 1 Sec Memory Limit: 512 Mb Submitted: 137 Solved: 37
Description
小L正在组织acm暑假集训,但众所周知,暑假集训的萌新中有OI神犇,也有暑假才开始学算法的萌新,如果统一集训的难度,无法很好地让萌新们得到训练,所以小L想了一个办法,根据每次测试的情况,改变萌新们的集训难度。现在将萌新们编号为1到n,最初萌新们的集训难度为v0,测试后有两种操作,第一种是某一区间的萌新的集训难度同时提高,另一种是将某一段区间的萌新的集训难度变为同一个数,同时,Wells希望在某次调整难度之后,知道某一段区间的萌新的集训难度之和,由于小L比较鶸,他并不知道如何快速解决这个问题,你能帮帮他嘛?
Input
第一行三个数n,m,v0 表示有n名萌新和m次调整,初始时全部萌新的集训难度都为v0
第2~m+1行 每行三个数或四个数
0 x y v 表示把 [x,y]区间内的萌新的集训难度都增加v
1 x y v 表示把 [x,y]区间内的萌新的集训难度都变为v
2 x y表示询问[x,y]区间内萌新的集训难度之和
0
Output
每个询问一行,输出答案
Sample Input
3 5 0
0 1 3 1
1 2 3 2
2 1 1
2 2 2
2 2 3
Sample Output
1
2
4
Author
lfw
除了区间询问、区间加减还有区间修改的线段树。类似只有加减的线段树,就是要多用一个tag存修改的值,并且在加减的时候判断修改值的变量tag2是不是初值,不是初值就加到tag2上面。用change函数处理tag的时候如果tag2不等于初值就要把子结点的tag1清零。
因为直接拿模板改的所以重复代码较多比较丑陋
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
struct node
{
int l,r;
long long sum,lazy,com;
}tree[100005<<2];
int n,m,v0;
const int MAXL=1e9;
void build(int k,int l,int r)//生成线段树
{
tree[k].l=l;
tree[k].r=r;
tree[k].com=MAXL;
tree[k].lazy=0;
tree[k].sum=1ll*(r-l+1)*v0;
if(l==r) return;
int mid=(l+r)>>1;
build(k<<1,l,mid);
build(k<<1|1,mid+1,r);
}
void fengxian(int k)//处理区间修改的tag2的函数
{
if(tree[k].l!=tree[k].r)
{
tree[k<<1].sum=1ll*(tree[k<<1].r-tree[k<<1].l+1)*tree[k].com;
tree[k<<1|1].sum=1ll*(tree[k<<1|1].r-tree[k<<1|1].l+1)*tree[k].com;
tree[k<<1].lazy=0;
tree[k<<1|1].lazy=0;
tree[k<<1].com=tree[k].com;
tree[k<<1|1].com=tree[k].com;
}
tree[k].com=MAXL;
}
void change(int k)//处理区间加减的tag1的函数
{
if(tree[k].l!=tree[k].r)
{
tree[k<<1].sum+=1ll*(tree[k<<1].r-tree[k<<1].l+1)*tree[k].lazy;
tree[k<<1|1].sum+=1ll*(tree[k<<1|1].r-tree[k<<1|1].l+1)*tree[k].lazy;
if(tree[k<<1].com<MAXL)
tree[k<<1].com+=tree[k].lazy;
else tree[k<<1].lazy+=tree[k].lazy;
if(tree[k<<1|1].com<MAXL)
tree[k<<1|1].com+=tree[k].lazy;
else tree[k<<1|1].lazy+=tree[k].lazy;
}
tree[k].lazy=0;
}
long long ques(int k,int l,int r)//区间询问
{
if(tree[k].com<MAXL)
fengxian(k);
if(tree[k].lazy)
change(k);
//cout<<'q'<<tree[k].l<<' '<<tree[k].r<<' '<<tree[k].sum<<endl;
long long sum,mid=(tree[k].l+tree[k].r)>>1;
if(tree[k].l==l&&tree[k].r==r)
return tree[k].sum;
if(r<=mid)
sum=ques(k<<1,l,r);
else if(l>mid)
sum=ques(k<<1|1,l,r);
else
{
sum=ques(k<<1,l,mid)+ques(k<<1|1,mid+1,r);
tree[k].sum=tree[k<<1].sum+tree[k<<1|1].sum;
}
return sum;
}
void add(int k,int l,int r,long long x)//区间加减
{
if(tree[k].lazy)
change(k);
if(tree[k].com<MAXL)
fengxian(k);
tree[k].sum+=1ll*(r-l+1)*x;
if(tree[k].l==l&&tree[k].r==r)
{
tree[k].lazy+=x;
return;
}
int mid=(tree[k].l+tree[k].r)>>1;
if(r<=mid)
add(k<<1,l,r,x);
else if(l>mid)
add(k<<1|1,l,r,x);
else
{
add(k<<1,l,mid,x);add(k<<1|1,mid+1,r,x);
}
tree[k].sum=tree[k<<1].sum+tree[k<<1|1].sum;
}
void shengping(int k,int l,int r,long long x)//区间修改
{
if(tree[k].lazy)
change(k);
if(tree[k].com<MAXL)
fengxian(k);
if(tree[k].l==l&&tree[k].r==r)
{
tree[k].sum=1ll*(r-l+1)*x;
tree[k].com=x;
tree[k<<1].lazy=0;
tree[k<<1|1].lazy=0;
//cout<<'s'<<tree[k].l<<' '<<tree[k].r<<' '<<tree[k].sum<<endl;
return;
}
int mid=(tree[k].l+tree[k].r)>>1;
if(r<=mid)
shengping(k<<1,l,r,x);
else if(l>mid)
shengping(k<<1|1,l,r,x);
else shengping(k<<1,l,mid,x),shengping(k<<1|1,mid+1,r,x);
tree[k].sum=tree[k<<1].sum+tree[k<<1|1].sum;
//cout<<'s'<<tree[k].l<<' '<<tree[k].r<<' '<<tree[k].sum<<endl;
}
int main(void)
{
while(~scanf("%d%d%d",&n,&m,&v0))
{
build(1,1,n);
for(int i=0;i<m;i++)
{
int x,y,mark;
long long v;
scanf("%d",&mark);
if(mark==0)
{
scanf("%d%d%lld",&x,&y,&v);
add(1,x,y,v);
}
else if(mark==1)
{
scanf("%d%d%lld",&x,&y,&v);
shengping(1,x,y,v);
}
else if(mark==2)
{
scanf("%d%d",&x,&y);
printf("%lld\n",ques(1,x,y));
}
}
}
return 0;
}