链接: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;
}