2018年东北农业大学春季校赛 L-wyh 【线段树】

最新推荐文章于 2022-01-15 19:56:01 发布

原创最新推荐文章于 2022-01-15 19:56:01 发布 · 202 阅读

1 ·

CC 4.0 BY-SA版权

线段树专栏收录该内容

43 篇文章

订阅专栏

本文介绍了一种使用线段树的数据结构解决动态数组中元素增删及查找问题的方法。具体应用场景为统计不同“萌”值的天鹅数量，在天鹅数量变化的同时能够快速定位到特定排名的天鹅“萌”值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述
你们wyh学长小时候住在河边，因为周围的生态环境非常好，所以经常会有天鹅浮在湖面上，每只天鹅都长得不一样，它们偶尔排成一排，偶尔分散开，偶尔也会去其他河畔，wyh学长为了统计它们的个数，编了一个程序赋予它们一个“萌”值，但是这些天鹅很不听话，一会儿会从别的地方游过来一两只，一会儿又会在统计过程中游走一两只，现在请你帮他完成统计任务。
输入描述:
共有T(T<=10)组数据，每组数据第一行为两个数 N, M (N，M <= 500000)，代表有N只天鹅和M次操作，接下来一行是N个数字，下面M行首先会输入一个字符串S，接着会有三类操作，如果S是“insert”，接着输入一个正整数a，代表插入一只“萌”值为a的天鹅，如果S是“delete”，接着输入一个正整数a，代表删除一只“萌”值为a的天鹅，如果S是“query”，接着输入一个正整数k，代表查询“萌”值第k大的天鹅。
萌值为[1,1000000000]，并且保证一定存在第k大
输出描述:
对应每次询问，输出询问结果。
示例1
输入

1
5 4
6 4 2 9 1
query 2
insert 7
delete 6
query 2
输出

6

思路与这篇链接相似，这里就不做详细介绍了；

#include<bits/stdc++.h>
using namespace std;
#define lson rt << 1, l, mid
#define rson rt << 1 | 1, mid + 1, r
#define mms(x, y) memset(x, y, sizeof x)
#define MAX 1000000
int tree[MAX << 2];
void PushUp(int rt)
{
    tree[rt] = tree[rt << 1] + tree[rt << 1 | 1];
}
void Update(int rt, int l, int r, int i, int v)
{
    if(l == r)
    {
        tree[rt] += v;
        return;
    }
    int mid = (l + r) >> 1;
    if(i <= mid)
        Update(lson, i, v);
    else
        Update(rson, i, v);
    PushUp(rt);
}
int Query(int rt, int l, int r, int k)
{
    if(l == r)
        return l;
    int mid = (l + r) >> 1;
    if(tree[rt << 1 | 1] >= k)
        return Query(rson, k);
    else
        return Query(lson, k - tree[rt << 1 | 1]);
}

int main()
{
    int N;
    scanf("%d", &N);
    while(N--)
    {
        int n, m;
        scanf("%d%d", &n, &m);
        mms(tree, 0);
        for(int i = 0, x; i < n; i++)
        {
            scanf("%d", &x);
            Update(1, 1, MAX, x, 1);
        }
        char s[30];
        for(int k = 0, x; k < m; k++)
        {
            scanf("%s%d", s, &x);
            if(s[0] == 'q')
                printf("%d\n", Query(1, 1, MAX, x));
            if(s[0] == 'i')
                Update(1, 1, MAX, x, 1);
            if(s[0] == 'd')
                Update(1, 1, MAX, x, -1);
        }
    }
    return 0;
}