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

分析

造一颗可持久化线段树，然后打个标记，dfs一下即可

代码

#include 

#define N 10000010
#define INF 0x7fffff
#define ll long long

int read()
{
    int x = 0, f = 1;
    char ch = getchar();
    while (ch < '0' || ch > '9') {if (ch == '-') f = -1; ch = getchar();}
    while (ch >= '0' && ch <= '9') {x = x * 10 + ch - '0'; ch = getchar();}
    return x * f;
}

struct NOTE
{
    int val;
    int ls,rs;
}t[N];

int tot;
int root;

int n;

ll ans;

void change(int &x,int a,int b,int l,int r,int now)
{
    if (l > r)
        return;
    if (!x)
    {
        x = ++tot;
    }
    if (l == a && r == b)
    {
        t[x].val = std::max(t[x].val, now);
        return;
    }
    int mid = (a + b) >> 1;
    if (r <= mid)
        change(t[x].ls, a, mid, l, r, now);
    else
        if (l > mid)
            change(t[x].rs, mid + 1, b, l, r, now);
        else
        {
            change(t[x].ls, a, mid, l, mid, now);
            change(t[x].rs, mid + 1, b, mid + 1, r, now);
        }
}

void dfs(int x,int l,int r,int now)
{
    if (!x)
        return;
    now = std::max(now,t[x].val);
    if (!t[x].ls && !t[x].rs)
    {
        ans += (ll)(r - l + 1) * now;
        return;
    }
    int mid = (l + r) >> 1;
    dfs(t[x].ls, l, mid, now);
    dfs(t[x].rs, mid + 1, r, now);
    if (!t[x].ls)
        ans += (ll)(mid - l + 1) * now;
    if (!t[x].rs)
        ans += (ll)(r - mid) * now;
}

int main()
{
    int n = read();
    for (int i = 1; i <= n ; i++)
    {
        int l = read(), r = read(), now = read();
        change(root, 0, INF, l, r - 1, now);
    }
    dfs(root, 0, INF, 0);
    printf("%lld\n",ans);
}