【Codeforces Round #179 (Div. 2) C. Greg and Array】差分/线段树

链接：

http://codeforces.com/problemset/problem/296/C

题意：

给你n个数，有m次操作，每次操作为区间加，现在有k次大操作，每次大操作执行m个操作的[l,r]区间,问k次大操作之后这n个数变成什么样？

做法：

第一种做法就是先对k次大操作进行差分，就知道每个小操作执行次数，之后线段树维护区间加即可。
第二种做法就是知道每个小操作执行次数之后，在原数组上再差分一次，就可以得到每个数加多少值，$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;
}