【UVa 12436 Rip Van Winkle's Code 】线段树lazy标记

UVA12436

题目大意：

给你不超过400000次操作，操作区间范围为[1,250000]，每次操作可以为：

1.A L R 表示[L,R]中从i=L开始，每个数依次增加i-L+1直到i=R。

2.B L R 表示[L,R]中从i=L开始，每个数依次增加R-i+1直到i=R。

3.C L R V 表示将[L,R]内的数全部设为V。

4.S L R 查询[L,R]内的数之和。

解法 我的做法是 维护一个 col[][0] 为正代表加几个当前数 比如 下标 5 这个叶子col[rt][0] = 3那么加15
col[][1] 为正代表加几个N-当前数+1 比如 下标 5 这个叶子col[rt][0] = 3那么加3*（N-5+1）；
以A 操作为例子
就是对 1 - y col[0]区间加个1
1 - x-1 col[0]区间减个1
x 为 3，y 为 5
加 1 代表贡献
1 2 3 4 5
减一代表贡献
-1 -2 0 0 0
这样就是 2 - 5 剩下了
3 4 5
我们区间减 2
就是 1 2 3
所以col[4]代表区间减
set数组代表区间置换 置换和加之间有逻辑性
先加再置换显然就不需要加了

/*
    if you can't see the repay
    Why not just work step by step
    rubbish is relaxed
    to ljq
*/
#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
#include <cmath>
#include <map>
#include <stack>
#include <set>
#include <sstream>
#include <vector>
#include <stdlib.h>
#include <algorithm>
using namespace std;

#define dbg(x) cout<<#x<<" = "<< (x)<< endl
#define dbg2(x1,x2) cout<<#x1<<" = "<<x1<<" "<<#x2<<" = "<<x2<<endl
#define dbg3(x1,x2,x3) cout<<#x1<<" = "<<x1<<" "<<#x2<<" = "<<x2<<" "<<#x3<<" = "<<x3<<endl
#define max3(a,b,c) max(a,max(b,c))
#define min3(a,b,c) min(a,min(b,c))

typedef pair<int,int> pll;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int _inf = 0xc0c0c0c0;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const ll _INF = 0xc0c0c0c0c0c0c0c0;
const ll mod =  (int)1e9+7;

ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
ll ksm(ll a,ll b,ll mod){int ans=1;while(b){if(b&1) ans=(ans*a)%mod;a=(a*a)%mod;b>>=1;}return ans;}
ll inv2(ll a,ll mod){return ksm(a,mod-2,mod);}
void exgcd(ll a,ll b,ll &x,ll &y,ll &d){if(!b) {d = a;x = 1;y=0;}else{exgcd(b,a%b,y,x,d);y-=x*(a/b);}}//printf("%lld*a + %lld*b = %lld\n", x, y, d);
const int MAX_N = 250025;
const int N = 250001;
namespace sgt
{
    #define mid ((l+r)>>1)
    ll s[MAX_N<<2],col[MAX_N<<2][3],Set[MAX_N<<2];
    void up(int rt)
    {
        s[rt] = s[rt<<1]+s[rt<<1|1];
    }
    void down(int rt,int l,int r)
    {
        if(Set[rt]!=inf)
        {
            Set[rt<<1] = Set[rt];
            Set[rt<<1|1] = Set[rt];
            s[rt<<1] = 1ll*(mid-l+1)*Set[rt];
            s[rt<<1|1] = 1ll*(r-mid)*Set[rt];
            for(int i = 0;i<3;++i)
                col[rt<<1][i] = col[rt<<1|1][i] = 0;
            Set[rt] = inf;
        }
        if(col[rt][0])
        {
            col[rt<<1][0]+=col[rt][0];
            col[rt<<1|1][0] += col[rt][0];
            s[rt<<1] += 1ll*(mid-l+1)*(l+mid)/2;
            s[rt<<1|1] += 1ll*(r-mid)*(mid+1+r)/2;
            col[rt][0] = 0;
        }
        if(col[rt][1])
        {
            col[rt<<1][1] += col[rt][1];
            col[rt<<1|1][1] += col[rt][1];
            s[rt<<1] += 1ll*(mid-l+1)*(N-mid+1+N-l+1)/2;
            s[rt<<1|1] += 1ll*(r-mid)*(N-mid+N-r+1)/2;
            col[rt][1] = 0;
        }
        if(col[rt][2])
        {
            col[rt<<1][2]+=col[rt][2];
            col[rt<<1|1][2] += col[rt][2];
            s[rt<<1] += 1ll*(mid - l +1) * col[rt][2];
            s[rt<<1|1] += 1ll*(r-mid)*col[rt][2];
            col[rt][2] = 0;
        }
    }
    void build(int rt,int l,int r)
    {
        col[rt][2] = col[rt][0] = col[rt][1] = 0;Set[rt] = inf;
        if(l==r)
        {
            s[rt] = 0;
            return ;
        }
        build(rt<<1,l,mid);
        build(rt<<1|1,mid+1,r);
    }
    void update(int rt,int l,int r,int x,int y,int flag ,int v)
    {
        if(x<=l&&r<=y)
        {
            if(flag==1)
            {
                col[rt][0]+=v;
                s[rt]+=1ll*v*(r-l+1)*(l+r)/2;
            }
            else if(flag==2)
            {
                col[rt][1]+=v;
                s[rt]+=1ll*v*(N-l+1+N-r+1)*(r-l+1)/2;
            }
            else if(flag==3)
            {
                Set[rt] = v;
                col[rt][2] = col[rt][0] = col[rt][1] = 0;
                s[rt] = 1ll*(r-l+1)*v;
            }
            else if(flag==4)
            {
                col[rt][2]+=v;
                s[rt]+=1ll*v*(r-l+1);
            }
            return ;
        }
        down(rt,l,r);
        if(x>mid) update(rt<<1|1,mid+1,r,x,y,flag,v);
        else if(y<=mid) update(rt<<1,l,mid,x,y,flag,v);
        else update(rt<<1,l,mid,x,y,flag,v),update(rt<<1|1,mid+1,r,x,y,flag,v);
        up(rt);
    }
    ll query(int rt,int l,int r,int x,int y)
    {
        if(x<=l&&r<=y)
        {
            return s[rt];
        }
        down(rt,l,r);
        if(x>mid) return query(rt<<1|1,mid+1,r,x,y);
        else if(y<=mid) return query(rt<<1,l,mid,x,y);
        else return query(rt<<1,l,mid,x,y)+query(rt<<1|1,mid+1,r,x,y);
    }
    #undef mid
}

int main()
{
    //ios::sync_with_stdio(false);
    //freopen("a.txt","r",stdin);
    //freopen("b.txt","w",stdout);
    int t,x,y,z;
    char str[10];
    sgt::build(1,1,N);
    scanf("%d",&t);
    while(t--)
    {
        scanf("%s%d%d",str,&x,&y);
        if(x>y) swap(x,y);
        if(str[0]=='A')
        {
            sgt::update(1,1,N,1,y,1,1);
            if(x>1) sgt::update(1,1,N,1,x-1,1,-1);
            sgt::update(1,1,N,x,y,4,-(x-1));
        }
        else if(str[0]=='B')
        {
            sgt::update(1,1,N,x,N,2,1);
            if(y<N) sgt::update(1,1,N,y+1,N,2,-1);
            sgt::update(1,1,N,x,y,4,-(N-y));
        }
        else if(str[0]=='C')
        {
           scanf("%d",&z);
           sgt::update(1,1,N,x,y,3,z);
        }
        else printf("%d\n",sgt::query(1,1,N,x,y));
    }
    //fclose(stdin);
    //fclose(stdout);
    //cout << "time: " << (long long)clock() * 1000 / CLOCKS_PER_SEC << " ms" << endl;
    return 0;
}