SP18185 GIVEAWAY - Give Away 题解

最短解？（格式化前）

需要维护带单点修改的二维偏序，考虑树套树。

手写太麻烦了，考虑树状数组套 __gnu_pbds::tree。

用 tree<T, null_type, less<T>, rb_tree_tag, tree_order_statistics_node_update> c 定义一棵元素类型为 T 的红黑树。

本题中需要用到的操作：

• c.size()：返回 c 的大小。
• c.insert(x)：在 c 中插入 x。
• c.erase(x)：在 c 中删除 x。
• c.order_of_key(x)：返回 c 中小于 x 的元素个数。
• c.lower_bound(x)：返回 c 中大于等于 x 的最小元素的迭代器。

但是 __gnu_pbds::tree 存不了重复元素，所以用 std::pair 区分重复元素。

然后正常地写树状数组套平衡树即可。

#include <cstdio>
#include <algorithm>
#include <ext/pb_ds/tree_policy.hpp>
#include <ext/pb_ds/assoc_container.hpp>
using namespace std;
using namespace __gnu_pbds;
int n, m, z, a[500050];
struct T
{
    tree<pair<int, int>, null_type, less<pair<int, int>>, rb_tree_tag, tree_order_statistics_node_update> c;
    void C(int x, int k)
    {
        if (k)
            c.insert({x, ++z});
        else
            c.erase(c.lower_bound({x, 1}));
    }
    int Q(int x) { return c.size() - c.order_of_key({x, 1}); }
} c[500050];
void C(int x, int y, int k)
{
    for (; x <= n; x += x & -x)
        c[x].C(y, k);
}
int Q(int x, int y, int s)
{
    int q = 0;
    for (--x; y > x; y &= y - 1)
        q += c[y].Q(s);
    for (; x > y; x &= x - 1)
        q -= c[x].Q(s);
    return q;
}
int main()
{
    scanf("%d", &n);
    for (int i = 1; i <= n; ++i)
        scanf("%d", a + i), C(i, a[i], 1);
    scanf("%d", &m);
    for (int i = 0, o, x, y, k; i < m; ++i)
    {
        scanf("%d%d%d", &o, &x, &y);
        if (o)
            C(x, a[x], 0), C(x, a[x] = y, 1);
        else
            scanf("%d", &k), printf("%d\n", Q(x, y, k));
    }
    return 0;
}