HDU 4302 Holedox Eating（线段树/优先队列/multiset）
题意：一条线段，在某些时刻上面的某些点会随机出现蛋糕。某种动物最初在0位置，在该动物想吃蛋时，它会跑到离他最近的蛋糕那里；如果有两个蛋糕与它的距离相同，就按照上次前进的方向走。输出它走过的总路程。
其实就是个水题，然而坑了一天，交了两页的记录 - - 写个博客纪念下
赛上用线段树区间最值过的，赛后看题解提到也可以线段树 + 二分做，又听说了 pq 和 multiset 的版本，就写了一发 pq，结果有个地方觉得写得不对但还是莫名AC了，觉得奇怪就坑了一下午 - - 还好最后终于找到了原因，并且用下面的数据叉掉了集训队里一个汉子的AC代码……这里output应该是26~
关于这个case和这个隐蔽修改方向导致WA的问题的解释以及multiset版本的代码，详见 lwt学长哒博客
总之，写代码前一定要思考清楚，要写出美观不易错的代码~
1
20 13
0 4
1
0 3
1
0 3
1
0 1
0 5
0 20
1
1
1
1 - 线段树区间最值
网上思路较多，就不细说了 - -
//赛上的线段树区间最值版
#include - 然后是二分版本哒~可以不记录区间最值，只要记录区间内cake数就可以了，但是要写两个query，不太好看。
思路是每次查询到某个区间时，如果当前区间内没有cake，直接返回。如果要查找区间最右边的值，就优先查找右子树，然后才是左子树，查找区间左值类比即可。
不过，网上虽然有人提了这个思路，但是几乎没有清晰代码，所以也贴下~
// 二分版本~
// 省略头文件50行
const int inf = 0x3f3f3f3f, maxn = 1e5 + 5;
int sum[maxn << 2];
void build(int id, int l, int r) {
sum[id] = 0;
if(l == r) return;
int m = l + r >> 1;
build(lson);
build(rson);
}
void update(int id, int l, int r, int pos, int d) {
sum[id] += d;
if(l == r) return;
int m = l + r >> 1;
if(pos <= m) update(lson, pos, d);
else update(rson, pos, d);
}
int queryl(int id, int l, int r, int a, int b) { // 查找区间最左边的cake的位置，没有则返回inf
if(l == r) return l;
int m = l + r >> 1;
int res = inf;
if(a <= m && sum[ls])
res = queryl(lson, a, b);
if(res == inf && b > m && sum[rs])
res = queryl(rson, a, b);
return res;
}
// 注意两个query的区别
int queryr(int id, int l, int r, int a, int b) { // 查找区间最右边的cake的位置，没有则返回 -inf
if(l == r) return l;
int m = l + r >> 1;
int res = -inf;
if(b > m && sum[rs])
res = queryr(rson, a, b);
if(res == -inf && a <= m && sum[ls])
res = queryr(lson, a, b);
return res;
}
ll solve(void) {
int n, q, a, b;
ll ans = 0;
int pre = 1, now = 1;
scanf("%d%d", &n, &q);
n++;
build(root);
while(q--) {
scanf("%d", &a);
if(!a) {
scanf("%d", &b);
update(root, b + 1, 1);
} else {
int goal, lef, rig;
lef = queryr(root, 1, now);
rig = queryl(root, now + 1, n);
if(now - lef != rig - now)
goal = now - lef < rig - now ? lef : rig;
else
goal = pre > 0 ? rig : lef;
if(abs(goal) == inf) continue;
GOTO: update(root, goal, -1);
ans += abs(goal - now);
if(goal > now) pre = 1;
else if(goal < now) pre = -1;
now = goal;
}
}
return ans;
}
int main(void) {
int t, kase = 0;
scanf("%d", &t);
while(t--)
printf("Case %d: %I64d\n", ++kase, solve());
return 0;
}- pq版本也来一发，网上代码也很多，不细说了 - -
不过说实话，用两个pq维护虽然听上去很机智，但是代码比较丑 也可能是我写的不好看吧 - - 感觉 multiset 或者 map 写会更简单
// pq 版~
const int inf = 0x3f3f3f3f, maxn = 1e5 + 5;
ll solve(void) {
priority_queue <int> pql;
priority_queue <int, vector<int>, greater<int> > pqr;
int n, q, a, b;
ll ans = 0;
int pre = 1, now = 0;
scanf("%d%d", &n, &q);
while(q--) {
scanf("%d", &a);
if(!a) {
scanf("%d", &b);
if(b >= now) pqr.push(b);
else pql.push(b);
} else {
int choice = 1, lef = -inf, rig = inf;
if(pql.size())
lef = pql.top();
if(pqr.size())
rig = pqr.top();
if(now - lef != rig - now) {
if(now - lef < rig - now)
choice = -1;
} else choice = pre;
if(choice > 0 && rig != inf) {
pqr.pop();
ans += rig - now;
if(rig != now)
pre = choice;
now = rig;
} else if(choice < 0 && lef != -inf) {
pql.pop();
ans += now - lef;
pre = choice;
now = lef;
}
}
}
return ans;
}
int main(void) {
int t, kase = 0;
scanf("%d", &t);
while(t--)
printf("Case %d: %I64d\n", ++kase, solve());
return 0;
}