dj 们在玩 k k k 个串的游戏
首先,牠们拿出了一个长度为 n n n 的数字序列,选出其中的一个连续子串,然后统计其子串中所有数字的和(重复出现的数字只被统计一次)
dj 们想知道,在这个数字序列所有连续的子串中,按照以上方式统计其所有数字的和,第 k k k 大的和是多少
对于 100 % 100\% 100% 的数据:满足 1 ≤ n ≤ 1 0 5 1 \leq n \leq 10^5 1≤n≤105, 1 ≤ k ≤ 2 × 1 0 5 1 \leq k \leq 2 \times 10^5 1≤k≤2×105, 0 ≤ ∣ a i ∣ ≤ 1 0 9 0 \leq |a_i| \leq 10^9 0≤∣ai∣≤109;并且保证存在第 k k k 大的和
常见 trick 题,枚举右端点 r r r。
然后有个无敌的做法,我考虑每次都取对于「某个序列」的最大值,那么无论如何我取 k k 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;
}