首页 > 其他好文 > 详细

主席树或者离散化+分块 BZOJ 4636

时间：2017-03-04 22:00:59 阅读：284 评论：0 收藏：0 [点我收藏+]

标签：data int des back void date bit ack str

4636: 蒟蒻的数列

Time Limit: 30 Sec Memory Limit: 256 MB
Submit: 381 Solved: 177
[Submit][Status][Discuss]

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

思路一：主席树（刚开始没想到，我好菜呀，明明这两天主席树还刚刚碰到的）

因为一共只有40000个操作，而且a和b的询问区间特别大，所以所有的更新节点不会太多，那么我们就用主席树，因为主席树每次最多就建立log个节点，所以能否符合要求。所以就用主席树打标记，然后如果lnub=0，那么就ans+=(mid-l+1)*val，如果rnub=0，那么ans+=(r-mid)*val，如果lnub=0&&rnub=0,那么ans+=(r-l+1)*val;即可完成query操作

//看看会不会爆int!数组会不会少了一维！
//取物问题一定要小心先手胜利的条件
#include <bits/stdc++.h>
using namespace std;
#pragma comment(linker,"/STACK:102400000,102400000")
#define LL long long
#define ALL(a) a.begin(), a.end()
#define pb push_back
#define mk make_pair
#define fi first
#define se second
#define haha printf("haha\n")
const int inf = 1e9 + 5;
const int maxn = 40000 + 5;
int n, root[maxn], tot;
struct Tree{
    int lnub, rnub, val;
}tree[maxn * 100];
LL ans;

int update(int o, int ql, int qr, int l, int r, int cost){
    int k = ++tot;
    tree[k] = tree[o];
    if (ql <= l && qr >= r) {
        tree[k].val = max(tree[k].val, cost);
        return k;
    }
    int mid = (l + r) / 2;
    if (ql <= mid) tree[k].lnub = update(tree[o].lnub, ql, qr, l, mid, cost);
    if (qr > mid) tree[k].rnub = update(tree[o].rnub, ql, qr, mid + 1, r, cost);
    return k;
}

void push_down(int o){
    int lb = tree[o].lnub, rb = tree[o].rnub;
    tree[lb].val = max(tree[lb].val, tree[o].val);
    tree[rb].val = max(tree[rb].val, tree[o].val);
}

void query(int o, int l, int r){
    if (!tree[o].lnub && !tree[o].rnub){
        ans = ans + 1LL * (r - l + 1) * tree[o].val;
        return ;
    }
    push_down(o);
    int mid = (l + r) / 2;
    if (tree[o].lnub) query(tree[o].lnub, l, mid);
    if (tree[o].rnub) query(tree[o].rnub, mid + 1, r);
    if (!tree[o].lnub) ans = ans + 1LL * (mid - l + 1) * tree[o].val;
    if (!tree[o].rnub) ans = ans + 1LL * (r - mid) * tree[o].val;
}

int main(){
    cin >> n;
    for (int i = 1; i <= n; i++){
        int a, b, k; scanf("%d%d%d", &a, &b, &k);
        root[i] = update(root[i - 1], a, b-1, 1, inf, k);
    }
    query(root[n], 1, inf);
    cout << ans << endl;
    return 0;
}