「bzoj 4504」K个串

dj 们在玩 $k$ 个串的游戏

首先，牠们拿出了一个长度为 $n$ 的数字序列，选出其中的一个连续子串，然后统计其子串中所有数字的和（重复出现的数字只被统计一次）

dj 们想知道，在这个数字序列所有连续的子串中，按照以上方式统计其所有数字的和，第 $k$ 大的和是多少
对于 $100\%$ 的数据：满足 $1\le n\le 10^5$，$1\le k\le 2\times 10^5$，$0\le |a_i|\le 10^9$；并且保证存在第 $k$ 大的和

常见 trick 题，枚举右端点 $r$。

然后有个无敌的做法，我考虑每次都取对于「某个序列」的最大值，那么无论如何我取 $k$ 次总能取出答案，那么考虑这个「某个序列」的取法。

一个比较 constructive 的做法是，先考虑所有前缀，然后把最大值的区间以最大值的位置分裂得到的两个区间加入 considerate。

考虑具体实现，弄个堆就行了。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define cmin(x, ...) x = min({x, __VA_ARGS__})
#define cmax(x, ...) x = max({x, __VA_ARGS__})
#define fors(i, l, r, ...) for(int i = (l), i##end = (r), ##__VA_ARGS__; i <= i##end; i++)
#define dfors(i, r, l, ...) for(int i = (r), i##end = (l), ##__VA_ARGS__; i >= i##end; i--)
int n, K, rt[100100];
map<int, int> lst;
struct _5 {
    ll v;
    int l, r, pos, x;
    bool operator<(const _5& o) const { return v < o.v; }
};
priority_queue<_5> q;
ll tag[7000000];
int ls[7000000], rs[7000000], _tt;
pair<ll, int> a[7000000];
int build(int l, int r) {
    int x = ++_tt;
    a[x] = {0, l};
    if(l == r) return x;
    int mid = (l+r)/2;
    ls[x] = build(l, mid), rs[x] = build(mid+1, r);
    return x;
}
int propagate(int x, const ll& v) {
    int y = ++_tt;
    ls[y] = ls[x], rs[y] = rs[x], a[y] = a[x], tag[y] = tag[x]+v;
    return a[y].first += v, y;
}
void push(int x) {
    if(tag[x] == 0ll) return;
    ls[x] = propagate(ls[x], tag[x]);
    rs[x] = propagate(rs[x], tag[x]);
    tag[x] = 0;
}
int impl(int p, int q, const ll& v, int x, int l, int r) {
    if(l > q || r < p) return x;
    if(l >= p && r <= q) return propagate(x, v);
    int y = ++_tt, mid = (l+r)/2;
    push(x);
    ls[y] = ls[x], rs[y] = rs[x], a[y] = a[x];
    ls[y] = impl(p, q, v, ls[x], l, mid);
    rs[y] = impl(p, q, v, rs[x], mid+1, r);
    return a[y] = max(a[ls[y]], a[rs[y]]), y;
}
pair<ll, int> get(int p, int q, int x, int l, int r) {
    if(l >= p && r <= q) return a[x];
    int mid = (l+r)/2;
    push(x);
    if(mid >= q) return get(p, q, ls[x], l, mid);
    else if(mid < p) return get(p, q, rs[x], mid+1, r);
    return max(get(p, q, ls[x], l, mid), get(p, q, rs[x], mid+1, r));
}
void decl(int l, int r, int x) {
    if(l > r) return;
    pair<ll, int> t = get(l, r, x, 1, n);
    q.push({t.first, l, r, t.second, x});
}
signed main() {
    ios::sync_with_stdio(0), cin.tie(0);
    cin >> n >> K;
    rt[0] = build(1, n);
    fors(i, 1, n, x) {
        cin >> x;
        rt[i] = impl(lst[x]+1, i, x, rt[i-1], 1, n);
        lst[x] = i, decl(1, i, rt[i]);
    }
    _5 t;
    while(K--) {
        t = q.top();
        q.pop();
        decl(t.l, t.pos-1, t.x), decl(t.pos+1, t.r, t.x);
    }
    cout << t.v << "\n";
    return 0;
}