洛谷 P2073 送花
Description
这些花都很漂亮，每朵花有一个美丽值W，价格为C。
小明一开始有一个空的花束，他不断地向里面添加花。他有以下几种操作：
操作 含义
1 W C 添加一朵美丽值为W，价格为C的花。
3 小明觉得当前花束中最便宜的一朵花太廉价，不适合送给小红，所以删除最便宜的一朵花。
2 小明觉得当前花束中最贵的一朵花太贵，他心疼自己的钱，所以删除最贵的一朵花。
-1 完成添加与删除，开始包装花束
若删除操作时没有花，则跳过删除操作。
如果加入的花朵价格已经与花束中已有花朵价格重复，则这一朵花不能加入花束。
请你帮小明写一个程序，计算出开始包装花束时，花束中所有花的美丽值的总和，以及小明需要为花束付出的总价格。
Input
- 若干行，每行一个操作，以-1结束。
Output
- 一行，两个空格隔开的正整数表示开始包装花束时，花束中所有花的美丽值的总和。以及小明需要为花束付出的总价格。
Sample Input
1 1 1 1 2 5 2 1 3 3 3 1 5 2 -1
Sample Output
8 5
Data Size
对于20%数据，操作数<=100，1<=W,C<=1000。
对于全部数据，操作数<=100000,1<=W,C<=1000000。
题解：
- 平衡树。
- 这题还算是比较模版吧。
- 我比较喜欢这题的操作2和3，可以用比较巧妙的方法解决。用自带的前驱后缀函数找到最大/小权值！找最大权值就找inf的前缀，找最小权值就找-inf的后缀。
- 这样就不用再开一个优先队列或者找第k大了！
- 发现没有，很多题使用平衡树都是在用其灵活的前缀/后缀函数！
#include <iostream>
#include <cstdio>
#include <cstdlib>
#define N 100005
#define inf 0x7fffffff
using namespace std;
struct T {int l, r, v1, v2, dat;} t[N];
int op, root, tot, ans1, ans2;
int read()
{
int x = 0; char c = getchar();
while(c < '0' || c > '9') c = getchar();
while(c >= '0' && c <= '9') {x = x * 10 + c - '0'; c = getchar();}
return x;
}
int New(int v1, int v2)
{
t[++tot].v1 = v1, t[tot].v2 = v2;
t[tot].dat = rand();
return tot;
}
void zig(int &y)
{
int x = t[y].l;
t[y].l = t[x].r, t[x].r = y, y = x;
}
void zag(int &x)
{
int y = t[x].r;
t[x].r = t[y].l, t[y].l = x, x = y;
}
void insert(int &p, int v1, int v2)
{
if(!p) {p = New(v1, v2); return;}
if(v1 == t[p].v1) return;
if(v1 < t[p].v1)
{
insert(t[p].l, v1, v2);
if(t[t[p].l].dat > t[p].dat) zig(p);
}
else
{
insert(t[p].r, v1, v2);
if(t[t[p].r].dat > t[p].dat) zag(p);
}
}
bool find(int p, int val)
{
if(!p) return 0;
if(t[p].v1 == val) return 1;
if(val < t[p].v1) return find(t[p].l, val);
else return find(t[p].r, val);
}
int preNext(int op, int val)
{
int ans = op == 0 ? 2 : 1, p = root;
while(p)
{
if(t[p].v1 == val)
{
if(!op && t[p].l)
{
p = t[p].l;
while(t[p].r) p = t[p].r;
ans = p; break;
}
else if(!op) break;
if(op && t[p].r)
{
p = t[p].r;
while(t[p].l) p = t[p].l;
ans = p; break;
}
else if(op) break;
}
if(!op && t[p].v1 < val && t[p].v1 > t[ans].v1) ans = p;
if(op && t[p].v1 > val && t[p].v1 < t[ans].v1) ans = p;
p = val < t[p].v1 ? t[p].l : t[p].r;
}
return t[ans].v1;
}
void erase(int &p, int val)
{
if(!p) return;
if(val == t[p].v1)
{
if(t[p].l || t[p].r)
{
if(!t[p].r || t[t[p].l].dat > t[t[p].r].dat) zig(p), erase(t[p].r, val);
else zag(p), erase(t[p].l, val);
}
else p = 0;
return;
}
val < t[p].v1 ? erase(t[p].l, val) : erase(t[p].r, val);
}
void dfs(int p)
{
if(!p) return;
if(t[p].v1 != inf && t[p].v1 != -inf) ans1 += t[p].v1;
if(t[p].v2 != inf && t[p].v2 != -inf) ans2 += t[p].v2;
dfs(t[p].l), dfs(t[p].r);
}
int main()
{
New(inf, inf), New(-inf, -inf), root = 1, t[1].l = 2;
while(scanf("%d", &op) == 1)
{
if(op == -1) break;
if(op == 1)
{
int bea = read(), cos = read();
if(!find(root, cos)) insert(root, cos, bea);
}
else if(op == 2)
{
int cos = preNext(0, inf);
if(cos != -inf) erase(root, cos);
}
else if(op == 3)
{
int cos = preNext(1, -inf);
if(cos != inf) erase(root, cos);
}
}
dfs(root);
cout << ans2 << ' ' << ans1 << endl;
return 0;
}