【线段树】295A Greg and Array

最新推荐文章于 2020-03-11 16:45:11 发布

CN_swords

最新推荐文章于 2020-03-11 16:45:11 发布

阅读量365

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构

本文链接：https://blog.youkuaiyun.com/qq_33199236/article/details/76422972

数据结构专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一种使用两个线段树的数据结构来解决序列更新与查询的问题。具体包括了如何通过线段树记录操作次数及操作值，并在一系列操作后快速获取更新后的序列值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Link：http://codeforces.com/contest/296/problem/C

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;

/*
题意：给你初始序列，m个操作，再给k个询问，每个询问的范围l，r，
表示操作l到r都执行一遍，问最后的序列是怎么样的。
题解：两个线段树维护，第一个先求出每个操作的操作次数，第二个知道每个的操作
次数，得每个操作总共加上的值，求序列的值
*/

const int Maxn = 1e5+6;
const LL mod = 1e9+7;
struct Node{
    int l,r;
    LL lazy,sum;
    void update(LL x){
        sum += (LL)(r-l+1)*x;
        lazy += x;
    }
}node[Maxn*4];
int a[Maxn];
void push_up(int x){
    node[x].sum = node[x<<1].sum+node[x<<1|1].sum;
}
void push_down(int x){
    LL lazyval = node[x].lazy;
    if(lazyval){
        node[x<<1].update(lazyval);
        node[x<<1|1].update(lazyval);
        node[x].lazy = 0;
    }
}
void build(int x,int l,int r){
    node[x].l = l; node[x].r = r;
    node[x].lazy = node[x].sum = 0;
    if(l == r){
        node[x].sum = a[l];
    }
    else{
        int mid = (l+r)>>1;
        build(x<<1,l,mid);
        build(x<<1|1,mid+1,r);
        push_up(x);
    }
}

void update(int x,int l,int r,LL v){
    int L = node[x].l, R = node[x].r;
    if(L >= l && r >= R){
        node[x].update(v);
    }
    else{
        push_down(x);
        int mid = (L+R)>>1;
        if(mid >= l)
            update(x<<1,l,r,v);
        if(r > mid)
            update(x<<1|1,l,r,v);
        push_up(x);
    }
}

LL query(int x,int l,int r){
    int L = node[x].l, R = node[x].r;
    if(L >= l && r >= R){
        return node[x].sum;
    }
    else{
        push_down(x);
        LL res = 0;
        int mid = (L+R)>>1;
        if(mid >= l)
            res += query(x<<1,l,r);
        if(r > mid)
            res += query(x<<1|1,l,r);
        push_up(x);
        return res;
    }
}
struct Op{
    int l,r;
    LL d;
}op[Maxn];
int main(){
    int n,m,k;
    scanf("%d%d%d",&n,&m,&k);
    memset(a,0,sizeof(a));
    build(1,1,m);
    for(int i = 1; i <= n; i++)
        scanf("%d",&a[i]);
    for(int i = 1; i <= m; i++)
        scanf("%d%d%I64d",&op[i].l,&op[i].r,&op[i].d);
    while(k--){
        int l,r;
        scanf("%d%d",&l,&r);
        update(1,l,r,1LL);
    }
    for(int i = 1; i <= m; i++){
        LL x = query(1,i,i);
        op[i].d = op[i].d*x;
    }
    build(1,1,n);
    for(int i = 1; i <= m; i++){
        update(1,op[i].l,op[i].r,op[i].d);
    }
    for(int i = 1; i <= n; i++){
        LL x = query(1,i,i);
        if(i == n)
            printf("%I64d\n",x);
        else
            printf("%I64d ",x);
    }
    return 0;
}