bzoj 4636: 蒟蒻的数列

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

 

题解：

 

应该是线段树吧。。

原来的写法行不通了，换了动态开节点。。

只要在树上打个标记，最后遍历一遍出解。。。

#include<stdio.h>
#include<iostream>
using namespace std;
const int N=400005;
#define ll long long
int n,i,x,y,z,rt,tot,t[N<<2],ls[N<<2],rs[N<<2];
ll ans;
inline void read(int &v){
    char ch,fu=0;
    for(ch='*'; (ch<'0'||ch>'9')&&ch!='-'; ch=getchar());
    if(ch=='-') fu=1, ch=getchar();
    for(v=0; ch>='0'&&ch<='9'; ch=getchar()) v=v*10+ch-'0';
    if(fu) v=-v;
}
inline void update(int l,int r,int x,int y,int z,int&p)
{
    if(!p) p=++tot;
    if(x<=l&&r<=y)
    {
        t[p]=max(t[p],z);
        return;
    }
    int mid=(l+r)>>1;
    if(x<=mid) update(l,mid,x,y,z,ls[p]);
    if(y>mid) update(mid+1,r,x,y,z,rs[p]);
}
inline void solve(int l,int r,int p)
{
    if(!ls[p]&&!rs[p])
    {
        ans+=(ll)(r-l+1)*t[p];
        return;
    }
    if(ls[p]) t[ls[p]]=max(t[ls[p]],t[p]);
    if(rs[p]) t[rs[p]]=max(t[rs[p]],t[p]);
    int mid=(l+r)>>1;
    if(ls[p]) solve(l,mid,ls[p]);
    if(rs[p]) solve(mid+1,r,rs[p]);
    if(!ls[p]) ans+=(ll)(mid-l+1)*t[p];
    if(!rs[p]) ans+=(ll)(r-mid)*t[p];
}
int main()
{
    read(n);
    for(i=1;i<=n;i++)
    {
        read(x),read(y),read(z);
        y--;
        if(x<=y) update(1,1000000000,x,y,z,rt);
    }
    solve(1,1000000000,rt);
    cout<<ans;
    return 0;
}