POJ3468：A Simple Problem with Integers （线段树||树状数组||Splay解决基本问题的效率对比）

You have N integers, A₁, A₂, ... , A_N. You need to deal with two kinds of operations. One type of operation is to add some given number to each number in a given interval. The other is to ask for the sum of numbers in a given interval.

Input

The first line contains two numbers N and Q. 1 ≤ N,Q ≤ 100000.
The second line contains N numbers, the initial values of A₁, A₂, ... , A_N. -1000000000 ≤ A_i ≤ 1000000000.
Each of the next Q lines represents an operation.
"C a b c" means adding c to each of A_a, A_a+1, ... , A_b. -10000 ≤ c ≤ 10000.
"Q a b" means querying the sum of A_a, A_a+1, ... , A_b.

Output

You need to answer all Q commands in order. One answer in a line.

Sample Input

10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4

Sample Output

4
55
9
15

You have N integers, A₁, A₂, ... , A_N. You need to deal with two kinds of operations. One type of operation is to add some given number to each number in a given interval. The other is to ask for the sum of numbers in a given interval.

Input

The first line contains two numbers N and Q. 1 ≤ N,Q ≤ 100000.
The second line contains N numbers, the initial values of A₁, A₂, ... , A_N. -1000000000 ≤ A_i ≤ 1000000000.
Each of the next Q lines represents an operation.
"C a b c" means adding c to each of A_a, A_a+1, ... , A_b. -10000 ≤ c ≤ 10000.
"Q a b" means querying the sum of A_a, A_a+1, ... , A_b.

Output

You need to answer all Q commands in order. One answer in a line.

Sample Input

10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4

Sample Output

4
55
9
15

题意：就是区间更改，区间求和。

思路：常规线段树，也可以树状态数组，也可以splay。

ps：我记得csl蔡神就是经常用splay来做线段树题，我最近发现这真是个好习惯。很大的好处就是写平衡树可以写得很灵活吧。

这里主要是都写一下，然后对比下效率。  也尽量以后多写splay。）

 体会：由于splay基本操作后都要把Now节点splay到根节点，可以从这里想办法优化。

比如，不要求在线的时候（比如区间第K大），可以向莫队一样，排序后再回答，这样，每次splay的高度会小一些。

其他优化就待续了，目前平衡树做得少。

树状数组：1969ms：

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn=;
ll a[maxn],b[maxn],c[maxn];
char opt[];int n,m;
int lowbit(int x) {return x&(-x);}
void add(ll *bb,ll *cc,int x,int val)
{
    ll tmp=x*val;
    for(int i=x;i<=n+;i+=lowbit(i))
      bb[i]+=val,cc[i]+=tmp;
}
ll query(ll *bb,ll *cc,int x)
{
    ll res=;
    for(int i=x;i;i-=lowbit(i)) res+=bb[i]; res*=(x+);
    for(int i=x;i;i-=lowbit(i)) res-=cc[i];
    return res+a[x];
}
int main()
{
    while(~scanf("%d%d",&n,&m)){
        for(int i=;i<=n;i++) scanf("%lld",&a[i]),a[i]+=a[i-];
        for(int i=;i<=n;i++) b[i]=c[i]=;
        while(m--){
            scanf("%s",opt); int x,y,z;
            if(opt[]=='Q')scanf("%d%d",&x,&y),printf("%lld\n",query(b,c,y)-query(b,c,x-));
            else scanf("%d%d%d",&x,&y,&z),add(b,c,x,z),add(b,c,y+,-z);
        }
    } return ;
}

线段树：2407ms：

#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long
const int maxn=;
int n,m;int a[maxn];
struct TREE
{
    ll sum[maxn<<];int lazy[maxn<<];
    void build(int Now,int l,int r)
    {
        lazy[Now]=;
        if(l==r) { sum[Now]=a[l]; return;}
        int Mid=(l+r)>>;
        build(Now<<,l,Mid);
        build(Now<<|,Mid+,r);
        pushup(Now);
    }
    void add(int Now,int l,int r,int x,int y,int val)
    {
        if(x<=l&&y>=r) { sum[Now]+=(ll)(r-l+)*val;lazy[Now]+=val; return ;}
        pushdown(Now,l,r);  int Mid=(l+r)>>;
        if(y<=Mid) add(Now<<,l,Mid,x,y,val);
        else if(x>Mid) add(Now<<|,Mid+,r,x,y,val);
        else add(Now<<,l,Mid,x,Mid,val),add(Now<<|,Mid+,r,Mid+,y,val);
        pushup(Now);
     }
     ll query(int Now,int l,int r,int x,int y)
     {
        if(x<=l&&y>=r)  return sum[Now];
        pushdown(Now,l,r); int Mid=(l+r)>>;
        if(y<=Mid) return query(Now<<,l,Mid,x,y);
        else if(x>Mid) return query(Now<<|,Mid+,r,x,y);
        else return query(Now<<,l,Mid,x,Mid)+query(Now<<|,Mid+,r,Mid+,y);
        pushup(Now);
    }
    void pushup(int Now) { sum[Now]=sum[Now<<]+sum[Now<<|];}
    void pushdown(int Now,int l,int r)
    {
        int Mid=(l+r)>>;
        lazy[Now<<]+=lazy[Now];sum[Now<<]+=(ll)(Mid-l+)*lazy[Now];
        lazy[Now<<|]+=lazy[Now];sum[Now<<|]+=(ll)(r-Mid)*lazy[Now];
        lazy[Now]=;
    }
}Tree;
int main()
{
     while(~scanf("%d%d",&n,&m)){
         for(int i=;i<=n;i++) scanf("%d",&a[i]);
         Tree.build(,,n);
         for(int i=;i<=m;i++){
                char opt[];int x,y,z;
                scanf("%s",opt);
                if(opt[]=='Q') scanf("%d%d",&x,&y),printf("%lld\n",Tree.query(,,n,x,y));
                else scanf("%d%d%d",&x,&y,&z),Tree.add(,,n,x,y,z);
         }
     } return ;
}

splay：3891ms。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long
const int maxn=;
int a[maxn];
struct Splay
{
    int ch[maxn][],sz[maxn],fa[maxn],rt,cnt;
    ll sum[maxn],key[maxn],lazy[maxn];
    void init()
    {
        rt=cnt=;
    }
    int get(int x)
    {
        return ch[fa[x]][]==x;
    }
    void pushdown(int Now)
    {
        if(!lazy[Now]) return ;
        int cl=ch[Now][],cr=ch[Now][];
        if(cl){
             sum[cl]+=sz[cl]*lazy[Now];
             lazy[cl]+=lazy[Now]; key[cl]+=lazy[Now];
        }
        if(cr){
             sum[cr]+=sz[cr]*lazy[Now];
             lazy[cr]+=lazy[Now]; key[cr]+=lazy[Now];
        }
        lazy[Now]=;
    }
    void update(int Now)
    {
        sum[Now]=key[Now]; sz[Now]=;
        if(ch[Now][]) sum[Now]+=sum[ch[Now][]],sz[Now]+=sz[ch[Now][]];
        if(ch[Now][]) sum[Now]+=sum[ch[Now][]],sz[Now]+=sz[ch[Now][]];
    }
    void rotate(int x)
    {
        int old=fa[x],fold=fa[old],opt=(ch[old][]==x);
        pushdown(old); pushdown(x);
        fa[ch[x][opt^]]=old; ch[old][opt]=ch[x][opt^];
        ch[x][opt^]=old; fa[old]=x; fa[x]=fold;
        if(!fold) rt=x;
        else ch[fold][ch[fold][]==old]=x;
        update(old); update(x);
    }
    void splay(int x,int y)
    {
        for(int f;(f=fa[x])!=y;rotate(x)){
            if(fa[f]!=y)
               rotate(get(x)==get(f)?f:x);
        }
        if(!y) rt=x;
    }
    int build(int L,int R)
    {
        if(L>R) return -;
        int Mid=(L+R)>>;
        if(!rt) rt=Mid;
        key[Mid]=a[Mid]; sum[Mid]=a[Mid]; sz[Mid]=;
        lazy[Mid]=fa[Mid]=ch[Mid][]=ch[Mid][]=;
        int cl=build(L,Mid-);
        if(cl!=-) {
            ch[Mid][]=cl; fa[cl]=Mid; sz[Mid]+=sz[cl];
        }
        int cr=build(Mid+,R);
        if(cr!=-){
            ch[Mid][]=cr; fa[cr]=Mid; sz[Mid]+=sz[cr];
        }
        update(Mid);
        return Mid;
    }
    ll query(int x,int y)
    {
        splay(x,); splay(y,x);
        return sum[ch[ch[rt][]][]];
    }
    void change(int x,int y,int z)
    {
        splay(x,); splay(y,x);
        int t=ch[ch[rt][]][];
        key[t]+=z; lazy[t]+=z;
        sum[t]+=sz[t]*z;
    }
}S;
int main()
{
    int N,Q,x,y,z; char opt[];
    while(~scanf("%d%d",&N,&Q)){
        S.init();
        for(int i=;i<=N;i++) scanf("%d",&a[i+]); a[]=a[N+]=;
        S.build(,N+);
        while(Q--){
            scanf("%s",opt);
            if(opt[]=='C'){
                scanf("%d%d%d",&x,&y,&z);
                S.change(x,y+,z);
            }
            else {
                scanf("%d%d",&x,&y);
                printf("%lld\n",S.query(x,y+));
            }
        }
    }
    return ;
}