其他分享
首页 > 其他分享> > HDOJ-4027(线段树+区间更新(每个节点更新的值不同))

HDOJ-4027(线段树+区间更新(每个节点更新的值不同))

作者:互联网

Can You answer these queries?

HDOJ-4027

//区间修改(不同于统一修改,这里是区间中每一项修改的值都不一样)和区间查询
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn=100005;
int n,m;
long long dur[maxn];
long long sums[maxn<<2];
void pushup(int id,int l,int r){
    int lc=id<<1;
    int rc=id<<1|1;
    sums[id]=sums[lc]+sums[rc];
}
void build(int id,int l,int r){
    if(l==r){
        sums[id]=dur[l];
        return;
    }
    int mid=(l+r)>>1;
    int lc=id<<1;
    int rc=id<<1|1;
    build(lc,l,mid);
    build(rc,mid+1,r);
    pushup(id,l,r);
}
void update(int id,int l,int r,int p,int q){
    if(sums[id]==r-l+1){//当结点覆盖的叶子结点全是1,也就是说sum等于覆盖的长度时返回,不用继续计算。
        return;
    }
    if(l==r){
        sums[id]=sqrt(sums[id]);
        return;
    }
    int mid=(l+r)>>1;
    if(p<=mid){
        update(id<<1,l,mid,p,q);
    }
    if(q>mid){
        update(id<<1|1,mid+1,r,p,q);
    }
    pushup(id,l,r);
}
long long query(int id,int l,int r,int p,int q){
    long long sum=0;
    if(p<=l&&q>=r){
        return sum=sums[id];
    }
    int mid=(l+r)>>1;
    int lc=id<<1;
    int rc=id<<1|1;
    if(p<=mid){
        sum+=query(lc,l,mid,p,q);
    }
    if(q>mid){
        sum+=query(rc,mid+1,r,p,q);
    }
    return sum;
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    int k=0;
    while(cin>>n){
        for(int i=1;i<=n;i++){
            cin>>dur[i];
        }
        build(1,1,n);
        cin>>m;
        cout<<"Case #"<<++k<<":"<<endl;
        for(int i=0;i<m;i++){
            int t,p,q;
            cin>>t>>p>>q;//不保证p<q
            int temp=p;
            p=min(p,q);
            q=max(temp,q);
            if(t==0){//update
                update(1,1,n,p,q);
            }else{//query
                cout<<query(1,1,n,p,q)<<endl;
            }
        }
        cout<<endl;
    }
    return 0;
}

标签:结点,int,sum,long,4027,更新,include,HDOJ
来源: https://www.cnblogs.com/GarrettWale/p/11448479.html