19thZUCCPC H.Distance(对顶堆)
作者:互联网
题目链接
每次给两个数\(l,r\),要我们选一个\(x\)使得\(\sum_{i = 1} ^ {k}dist(l,x,r)\)的值最小。可以知道要让答案最小的话,\(x\)一定要刚好把\(l,r\)平均分在两边,如果不是平分的话,\(x\)就会向多的一侧移动,所以要我们求得就是中位数。
\(nbsp;\)emsp;中位数我们可以用对顶堆来动态维护。用一个大根堆来存储小的值,小根堆存放大的值,在有新的元素插入得时候,就可以将大根堆得堆顶元素顶到小根堆中,也可以将小根堆中多余得值顶回大根堆中。
void solve() {
int n; iocin >> n;
std::priority_queue<i64,V<i64>, std::greater<i64>> q2;
std::priority_queue<i64> q1;
i64 res = 0, sum1 = 0, sum2 = 0;
rep(i,0,n) {
i64 l, r;
iocin >> l >> r;
res -= (r - l);
q1.push(l), q1.push(r);
sum1 += l + r;
sum1 -= q1.top();
sum2 += q1.top();
q2.push(q1.top());
q1.pop();
i64 u = q1.top(), v = q2.top();
q1.pop(), q2.pop();
sum1 -= u, sum2 -= v;
sum1 += std::min(u, v), sum2 += std::max(u, v);
q1.push(std::min(u, v)), q2.push(std::max(u, v));
printf("%lld\n", (sum2 - sum1 + res) / 2);
}
}
标签:q1,Distance,q2,top,sum2,sum1,std,19thZUCCPC 来源: https://www.cnblogs.com/Haven-/p/16276949.html