洛谷 P2073 送花

洛谷 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;
}

转载于:https://www.cnblogs.com/BigYellowDog/p/11324369.html