一维树桩数组区间更新、区间查询

链接:https://www.nowcoder.com/acm/contest/135/I
来源:牛客网
 

区间 (interval)

时间限制:C/C++ 2秒,其他语言4秒
空间限制:C/C++ 32768K,其他语言65536K
64bit IO Format: %lld

题目描述

    Apojacsleam喜欢数组。

    他现在有一个n个元素的数组a,而他要对a[L]-a[R]进行M次操作:

        操作一:将a[L]-a[R]内的元素都加上P

        操作二:将a[L]-a[R]内的元素都减去P

    最后询问a[l]-a[r]内的元素之和?

    请认真看题干及输入描述。

输入描述:

输入共M+3行:

第一行两个数,n,M,意义如“题目描述”

第二行n个数,描述数组。

第3-M+2行,共M行,每行四个数,q,L,R,P,若q为1则表示执行操作2,否则为执行操作1

第4行,两个正整数l,r

输出描述:

一个正整数,为a[l]-a[r]内的元素之和

示例1

输入

复制

10 5
1 2 3 4 5 6 7 8 9 10
1 1 5 5
1 2 3 6
0 2 5 5 
0 2 5 8
1 4 9 6
2 7

输出

复制

23

说明

 
 
#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
#include<queue>
#include<set>
const int MAXN=1e6+10;
using namespace std;
int lowbit(int x){return x&(-x);}
int n,Q,opt,x,y;
long long A[MAXN],C[MAXN],sum[MAXN],z;
void modifyA(int x,int val){
    while(x<=n){
        A[x]+=val;
        x+=lowbit(x);
    }
}
void modifyC(int x,long long val){
    while(x<=n){
        C[x]+=val;
        x+=lowbit(x);
    }
}
void update(int from,int to,long long val){
    modifyA(from,val);
    modifyA(to+1,-val);
    modifyC(from,val*from);
    modifyC(to+1,-val*(to+1));
}
long long queryA(int loc){
    long long ans=0;
    while(loc){
        ans+=A[loc];
        loc-=lowbit(loc);
    }
    return ans;
}
long long queryC(int loc){
    long long ans=0;
    while(loc){
        ans+=C[loc];
        loc-=lowbit(loc);
    }
    return ans;
}
long long query(int loc)
 {
    return sum[loc]+(loc+1)*queryA(loc)-queryC(loc);
}
long long query(int from,int to)
{
    return query(to)-query(from-1);
}
int m;
int main()
{
    scanf("%d%d",&n,&m);
    for(register int i=1;i<=n;i++)scanf("%lld",&sum[i]),sum[i]+=sum[i-1];
    while(m--){
        scanf("%d%d%d",&opt,&x,&y);
        if(opt==1)
        {
            scanf("%lld",&z);
            update(x,y,-z);
        }
        else
        {
            scanf("%lld",&z);
            update(x,y,z);
        }
    }
    int L,R;
    scanf("%d%d",&L,&R);
    printf("%lld\n",query(L,R));
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_37891604/article/details/81164927