区间 (interval)

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

题目描述

    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

说明

思路：一般来说这种题会直接想起线段树或者树状数组。我第一次用的线段树，因为开4倍内存，超了；改用树状数组。套模板即可。

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 1000002   //记得多一点点就行，不然超内存
#define ll long long
using namespace std;
ll c1[N];
ll c2[N];
ll ans[N];
ll n,m;
ll lowbit(ll x)
{
    return x&(-x);
}
void update(ll x,ll val,ll *c)
{
    while(x<=n)
    {
        c[x]+=val;
        x+=lowbit(x);
    }
}
ll getsum(ll x,ll *c)
{
    ll s=0;
    while(x>0)
    {
        s+=c[x];
        x-=lowbit(x);
    }
    return s;
}
int main()
{
    memset(c1,0,sizeof c1);
    memset(c2,0,sizeof c2);
    memset(ans,0,sizeof ans);
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&ans[i]);
        ans[i]+=ans[i-1];
    }
    for(int i=1;i<=m;i++)
    {
        int op;
        cin>>op;
        if(op!=1)
        {
            ll s1,s2,s3;
            scanf("%lld%lld%lld",&s1,&s2,&s3);
            update(s1,s3,c1);
            update(s2+1,-s3,c1);
            update(s1,s1*s3,c2);
            update(s2+1,-(s2+1)*s3,c2);
        }
        else
        {
            ll s1,s2,s3;
            scanf("%lld%lld%lld",&s1,&s2,&s3);
            update(s1,-s3,c1);
            update(s2+1,s3,c1);
            update(s1,-s1*s3,c2);
            update(s2+1,(s2+1)*s3,c2);
        }
    }
    ll s1,s2;
    scanf("%lld%lld",&s1,&s2);
    ll cur=ans[s2]-ans[s1-1];
    cur+=getsum(s2,c1)*(s2+1)-getsum(s2,c2);
    cur-=getsum(s1-1,c1)*(s1)-getsum(s1-1,c2);
    printf("%lld\n",cur);
    return 0;
}

 

我们可以发现，在l到r的区间修改中，不管怎么改，a[i]-a[i-1]的值是不变的，所以我们只要处理差值就行。所以得到方法二：

#include<iostream>
#include <cstdio>
using namespace std;
 
typedef long long ll;
int const SIZE=1000002;
 
ll a[SIZE];
 
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i)
        scanf("%lld",&a[i]);
 
    for(int i=n;i>1;--i)
        a[i]=a[i]-a[i-1]; //计算后一项与前一项的差值
    int q,l,r,p;
    while(m--)
    {
        scanf("%d%d%d%d",&q,&l,&r,&p);
        if(q==1)
            a[l]-=p,a[r+1]+=p;  //因为中间的不变所以只要修改l，与r+1
        else
            a[l]+=p,a[r+1]-=p;
    }
    scanf("%d%d",&l,&r);
    ll sum=0;
    for(int i=2;i<=n;++i)
        a[i]=a[i]+a[i-1];
    for(int i=l;i<=r;++i)
        sum+=a[i];
    printf("%lld\n",sum);
    return 0;
}