XDU 挖掘机线段树求解

最新推荐文章于 2024-12-10 09:23:43 发布

原创最新推荐文章于 2024-12-10 09:23:43 发布 · 605 阅读

0 ·

CC 4.0 BY-SA版权

线段树专栏收录该内容

5 篇文章

订阅专栏

本文通过一个具体的题目实例，详细介绍了如何使用线段树解决区间更新与查询的问题，并提供了完整的代码实现，帮助读者理解线段树的应用场景及其实现细节。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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