UVA 12436-Rip Van Winkle's Code（线段树的区间更新）

题意：

long long data[250001];

void A( int st, int nd ) { for( int i = st; i \le nd; i++ ) data[i] = data[i] + (i - st + 1); }

void B( int st, int nd ) { for( int i = st; i \le nd; i++ ) data[i] = data[i] + (nd - i + 1); }

void C( int st, int nd, int x ) { for( int i = st; i \le nd; i++ ) data[i] = x; }

long long S( int st, int nd ) { long long res = 0; for( int i = st; i \le nd; i++ ) res += data[i]; return res; }

四个函数，对应四种操作Ａ、B、C、S给出m个操作，求s操作时返回的结果。

分析：

明显的线段树的区间更新，但懒惰标记的选择要想一想，C这个操作好处理、我们观察A、B发现都是加上一个等差数列、等差数列加等差数列还是等差数列、那我们选择等差数列的首项和公差做标记。

#include <map>
#include <set>
#include <list>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <vector>
#include <string>
#include <cctype>
#include <complex>
#include <cassert>
#include <utility>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
typedef pair<int,int> PII;
typedef long long ll;
#define lson l,m,rt<<1
#define pi acos(-1.0)
#define rson m+1,r,rt<<1|1
#define All 1,N,1
#define read freopen("in.txt", "r", stdin)
#define N 250001
const ll  INFll = 0x3f3f3f3f3f3f3f3fLL;
const int INF= 0x7ffffff;
const int mod =  ;
struct node{
    ll a1,d,sum,setv;
    int l,r;
}t[N<<];
void pushup(int rt){
    t[rt].sum=t[rt<<].sum+t[rt<<|].sum;
}
void pushdown(int rt){
    if(t[rt].setv!=INF){
        t[rt<<].setv=t[rt<<|].setv=t[rt].setv;
        t[rt<<].a1=t[rt<<].d=t[rt<<|].a1=t[rt<<|].d=;
        t[rt<<].sum=1LL*(t[rt<<].r-t[rt<<].l+)*t[rt].setv;
        t[rt<<|].sum=1LL*(t[rt<<|].r-t[rt<<|].l+)*t[rt].setv;
        t[rt].setv=INF;
    }
    if(t[rt].a1||t[rt].d){
        ll len1=t[rt<<].r-t[rt<<].l+;
        ll len2=t[rt<<|].r-t[rt<<|].l+;
        ll tmp1=t[rt].a1;
        ll tmp2=tmp1+len1*t[rt].d;
        t[rt<<].a1+=tmp1;
        t[rt<<|].a1+=tmp2;
        t[rt<<].d+=t[rt].d;
        t[rt<<|].d+=t[rt].d;
        t[rt<<].sum+=tmp1*len1+len1*(len1-)/*t[rt].d;//等差数列求和
        t[rt<<|].sum+=tmp2*len2+len2*(len2-)/*t[rt].d;
        t[rt].a1=t[rt].d=;
    }
}
void build(int l,int r,int rt){
    t[rt].l=l;
    t[rt].r=r;
    t[rt].sum=t[rt].a1=t[rt].d=;
    t[rt].setv=INF;
    if(l==r)return;
    int m=(l+r)>>;
    build(lson);
    build(rson);
   pushup(rt);
}
void update_add(int l,int r,int rt,ll d){
    if(l<=t[rt].l&&r>=t[rt].r){
        ll a1=(d==)?(t[rt].l-l+):(r-t[rt].l+);//确定首项
        ll len=t[rt].r-t[rt].l+;
         t[rt].a1+=a1;
         t[rt].d+=d;
         t[rt].sum+=a1*len+len*(len-)/*d;
         return;
    }
    pushdown(rt);
    int m=(t[rt].l+t[rt].r)>>;
    if(l<=m)update_add(l,r,rt<<,d);
    if(r>m)update_add(l,r,rt<<|,d);
    pushup(rt);
}
void update_set(int l,int r,int rt,ll x){
    if(l<=t[rt].l&&r>=t[rt].r){
        t[rt].setv=x;
        t[rt].a1=t[rt].d=;
        t[rt].sum=1LL*(t[rt].r-t[rt].l+)*x;
        return;
    }
    pushdown(rt);
    int m=(t[rt].l+t[rt].r)>>;
    if(l<=m)update_set(l,r,rt<<,x);
    if(r>m)update_set(l,r,rt<<|,x);
    pushup(rt);
}
ll query(int l,int r,int rt){
    if(l<=t[rt].l&&r>=t[rt].r){
        return t[rt].sum;
    }
    pushdown(rt);
    ll num=;
    int m=(t[rt].l+t[rt].r)>>;
    if(l<=m)num+=query(l,r,rt<<);
    if(r>m)num+=query(l,r,rt<<|);
    return num;
}
int main()
{
    int n,tll,trr;
    ll x;
    char op[];
     build(,N,);
    scanf("%d",&n);
        while(n--){
            scanf("%s%d%d",op,&tll,&trr);
            if(op[]=='A'){
                update_add(tll,trr,,);
            }
            else if(op[]=='B')
                update_add(tll,trr,,-);
            else if(op[]=='C'){
                scanf("%lld",&x);
                update_set(tll,trr,,x);
            }
            else if(op[]=='S'){
                printf("%lld\n",query(tll,trr,));
            }
        }
return ;
}