CSU-2151 集训难度

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;
}