题目链接: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;
}