POJ3468 树状数组的区间更新和区间求和

这道题由于数据较大,不能简单地使用一般的树状数组单点更新和单点向下求和,而应该使用区间范围的更新和求和。

不过区间求和的公式证明看得有点久,但最终还是弄懂了,以下为链接:点击打开链接


知道这个以后直接用就好了,不过要注意更新和查询时的左右边界的取值。


#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=400000+50;
long long n,m,a[maxn],c1[maxn],c2[maxn],sum[maxn];
char str[3];
long long Lowbit(long long x)
{
    return x&(-x);
}

void Add(long long a[],long long x,long long d)
{
    while(x<=n)
    {
        a[x]+=d;
        x+=Lowbit(x);
    }
}

long long Sum(long long a[],long long x)
{
    long long sum=0;
    while(x>0)
    {
        sum+=a[x];
        x-=Lowbit(x);
    }
    return sum;
}

int main()
{
    while(scanf("%lld%lld",&n,&m)!=EOF)
    {
        memset(c1,0,sizeof(c1));
        memset(c2,0,sizeof(c2));
        memset(sum,0,sizeof(sum));
        for(int i=1;i<=n;i++)
            scanf("%lld",&a[i]);
        sum[0]=0;
        for(int j=1;j<=n;j++)
            sum[j]=sum[j-1]+a[j];
        for(int i=1;i<=m;i++)
        {
            scanf("%s",str);
            if(str[0]=='Q')
            {
                long long left,right,insuml,insumr,insum;
                scanf("%lld%lld",&left,&right);
                insuml=sum[left-1]+(left)*Sum(c1,left-1)-Sum(c2,left-1);
                insumr=sum[right]+(right+1)*Sum(c1,right)-Sum(c2,right);
                insum=insumr-insuml;
                printf("%lld\n",insum);
            }
            else
            {
                long long left,right,addVal;
                scanf("%lld%lld%lld",&left,&right,&addVal);
                Add(c1,left,addVal);
                Add(c1,right+1,-addVal);
                Add(c2,left,left*addVal);
                Add(c2,right+1,-(right+1)*addVal);
            }
        }
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值