XDU 挖掘機 線段樹求解

題目鏈接：http://acm.xidian.edu.cn/problem.php?cid=1002&pid=1

看到別人有O（N)的解法（不會），不過這裏用線段樹，練習線段樹。

用線段樹維護變量：於最後我們只需要知道每一個點的修改次數，所以，我們只要記錄每個區間被修改的次數就好了，最後查詢時，在從上到下記錄下修改的值就好了。

代碼：

#include<iostream>
#include<cstdio>
#define LL long long
#define maxn 100010

using namespace std;
struct node{
    int l,r,k;
}T[maxn*3];

int n,m,A[maxn];

void build(int L,int R,int o){
    T[o].l=L;T[o].r=R;T[o].k=0;
    if(T[o].l==T[o].r) return;
    build(L,(L+R)/2,o*2);build((L+R)/2+1,R,o*2+1);
}

void add(int L,int R,int val,int o){
    if(T[o].l>=L && T[o].r<=R){
        T[o].k+=val;return;
    }
    if(T[o].l==T[o].r) return;
    int tl=o*2,tr=o*2+1;
    if(L>=T[tr].l) add(L,R,val,tr);
    else if(R<=T[tl].r) add(L,R,val,tl);
    else{
        add(L,T[tl].r,val,tl);
        add(T[tr].l,R,val,tr);
    }
}

int query(int t,int val,int o){//查詢每個點，val記錄該點的變化值
    if(T[o].l==T[o].r && T[o].l==t) return val+T[o].k;
    if(T[o].l==T[o].r) return 0;
    if(t>=T[o*2+1].l) return query(t,val+T[o].k,o*2+1);
    return query(t,val+T[o].k,o*2);
}

int main(){
    //freopen("D:\\in.txt","r",stdin);
    while(cin>>n){
        for(int i=1;i<=n;i++)
            scanf("%d",A+i);
        build(1,n,1);
        cin>>m;
        int l,r,k;
        for(int i=0;i<m;i++){
            scanf("%d %d %d",&l,&r,&k);
            add(l+1,r,0-k,1);
        }
        LL sum=0;
        for(int i=1;i<=n;i++){
            k=query(i,0,1);sum=sum+A[i]+k;A[i]+=k;
        }
        LL ave=sum/n;
        LL ans=0;
        for(int i=1;i<=n;i++)
            ans=ans+(LL)(A[i]-ave)*(A[i]-ave);
        cout<<ans<<endl;
    }
    return 0;
}