BZOJ_4636_蒟蒻的数列_线段树+动态开点

BZOJ_4636_蒟蒻的数列_线段树+动态开点

Description

蒟蒻DCrusher不仅喜欢玩扑克，还喜欢研究数列

题目描述

DCrusher有一个数列，初始值均为0，他进行N次操作，每次将数列[a,b)这个区间中所有比k小的数改为k，他想知

道N次操作后数列中所有元素的和。他还要玩其他游戏，所以这个问题留给你解决。

Input

第一行一个整数N，然后有N行，每行三个正整数a、b、k。

N<=40000 , a、b、k<=10^9

Output

一个数，数列中所有元素的和

Sample Input

4
2 5 1
9 10 4
6 8 2
4 6 3

Sample Output

16

---

把操作离线并从小到大排序，相当于线段树维护区间赋值操作。

本题可以离散化也可以动态开点，我这里处理不好线段树维护离散后的区间因此写了动态开点。

注意pushdown操作时如果没有这个儿子也要开出来。

 

代码：

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <stdlib.h>
using namespace std;
#define N 40050
#define maxn 1000000000
typedef long long ll;
struct A {
    int l,r,k;
}a[N];
bool cmp(const A &x,const A &y) {return x.k<y.k;}
int tag[N*80],cnt,n,ls[N*80],rs[N*80];
ll t[N*80];
/*int make(int x) {
    return lower_bound(v+1,v+cnt+1,x)-v;
}*/
void pushdown(int l,int r,int &p,int w) {
    if(!p) p=++cnt;
    t[p]=1ll*w*(r-l+1); tag[p]=w;
}
void update(int l,int r,int x,int y,int w,int &p) {
    if(p==0) p=++cnt;
    if(x<=l&&y>=r) {
        t[p]=1ll*w*(r-l+1);
        tag[p]=w;
        return ;
    }
    int mid=(l+r)>>1;
    if(tag[p]) {
        pushdown(l,mid,ls[p],tag[p]);
        pushdown(mid+1,r,rs[p],tag[p]);
        tag[p]=0;
    }
    if(x<=mid) update(l,mid,x,y,w,ls[p]);
    if(y>mid) update(mid+1,r,x,y,w,rs[p]);
    t[p]=t[ls[p]]+t[rs[p]];
}
int main() {
    scanf("%d",&n);
    int i;
    for(i=1;i<=n;i++) {
        scanf("%d%d%d",&a[i].l,&a[i].r,&a[i].k);
        a[i].r--;
        /*v[++cnt]=a[i].l;
        v[++cnt]=a[i].r;*/
    }
    /*sort(v+1,v+cnt+1);
    cnt=unique(v+1,v+cnt+1)-v-1;
    printf("%d\n",cnt);
    for(i=1;i<=cnt;i++) printf("%d\n",v[i]); */
    sort(a+1,a+n+1,cmp);
    int root=0;
    for(i=1;i<=n;i++) {
        if(a[i].k==0) continue;
        /*printf("%d %d\n",make(a[i].l),make(a[i].r));
        update(1,cnt,make(a[i].l),make(a[i].r),a[i].k,1);
        printf("sum=%lld\n",t[1]);*/
        update(1,maxn,a[i].l,a[i].r,a[i].k,root);
    }
    printf("%lld\n",t[1]);
}