链接:
http://codeforces.com/problemset/problem/296/C
题意:
给你n
个数,有m
次操作,每次操作为区间加,现在有k
次大操作,每次大操作执行m
个操作的[l,r]
区间,问k次大操作之后这n个数变成什么样?
做法:
第一种做法就是先对k次大操作进行差分,就知道每个小操作执行次数,之后线段树维护区间加即可。
第二种做法就是知道每个小操作执行次数之后,在原数组上再差分一次,就可以得到每个数加多少值,
O
(
n
)
O(n)
O(n)即可完成。
代码:
做法一:
#include<stdio.h>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn = 2e5+5;
struct T
{
int l,r,mid;
ll val,add;
}tree[maxn<<2];
int a[maxn];
void up(int rt)
{
tree[rt].val=tree[rt<<1].val+tree[rt<<1|1].val;
}
void down(int rt)
{
if(tree[rt].add)
{
ll tmp=tree[rt].add;
tree[rt<<1].add+=tmp;
tree[rt<<1].val+=1LL*(tree[rt<<1].r-tree[rt<<1].l+1)*tmp;
tree[rt<<1|1].add+=tmp;
tree[rt<<1|1].val+=1LL*(tree[rt<<1|1].r-tree[rt<<1|1].l+1)*tmp;
tree[rt].add=0;
}
}
void build(int rt,int l,int r)
{
tree[rt].l=l;
tree[rt].r=r;
tree[rt].add=0;
if(l==r)
{
tree[rt].val=a[l];
return ;
}
int mid=tree[rt].mid=l+r>>1;
build(rt<<1,l,mid);
build(rt<<1|1,mid+1,r);
up(rt);
}
void update(int rt,int l,int r,ll val)
{
if(tree[rt].l>r||tree[rt].r<l) return ;
if(tree[rt].l>=l&&tree[rt].r<=r)
{
tree[rt].add+=val;
tree[rt].val+=1LL*(tree[rt].r-tree[rt].l+1)*val;
return ;
}
down(rt);
if(tree[rt].mid>=l) update(rt<<1,l,r,val);
if(tree[rt].mid<r) update(rt<<1|1,l,r,val);
up(rt);
return ;
}
ll query(int rt,int pos)
{
if(tree[rt].l==tree[rt].r) return tree[rt].val;
down(rt);
if(pos<=tree[rt].mid) return query(rt<<1,pos);
else return query(rt<<1|1,pos);
}
int l[maxn],r[maxn],val[maxn],cnt[maxn];
int tt[maxn];
int main()
{
int n,m,k;
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
build(1,1,n);
for(int i=1;i<=m;i++) scanf("%d%d%d",&l[i],&r[i],&val[i]);
for(int i=1;i<=k;i++)
{
int u,v;
scanf("%d%d",&u,&v);
tt[u]++;
tt[v+1]--;
}
int tmp=0;
for(int i=1;i<=m;i++)
{
tmp+=tt[i];
cnt[i]=tmp;
}
for(int i=1;i<=m;i++) update(1,l[i],r[i],1LL*cnt[i]*val[i]);
for(int i=1;i<=n;i++) printf("%lld ",query(1,i));
return 0;
}
做法二:
#include<stdio.h>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn = 2e5+5;
int a[maxn];
int l[maxn],r[maxn],val[maxn],cnt[maxn];
int tt[maxn];
ll sum[maxn];
int main()
{
int n,m,k;
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
for(int i=1;i<=m;i++) scanf("%d%d%d",&l[i],&r[i],&val[i]);
for(int i=1;i<=k;i++)
{
int u,v;
scanf("%d%d",&u,&v);
tt[u]++;
tt[v+1]--;
}
int tmp=0;
for(int i=1;i<=m;i++)
{
tmp+=tt[i];
cnt[i]=tmp;
}
ll tp=0;
for(int i=1;i<=m;i++)
{
sum[l[i]]+=1LL*cnt[i]*val[i];
sum[r[i]+1]-=1LL*cnt[i]*val[i];
}
for(int i=1;i<=n;i++)
{
tp+=sum[i];
printf("%lld ",a[i]+tp);
}
return 0;
}