HDU2852 大于a的第k小数-树状数组

最新推荐文章于 2022-04-11 02:09:01 发布

Sky-J

最新推荐文章于 2022-04-11 02:09:01 发布

阅读量560

点赞数

分类专栏： ACM-树

本文链接：https://blog.csdn.net/sky_j123/article/details/21241871

版权

ACM-树专栏收录该内容

21 篇文章 0 订阅

订阅专栏

//sum(a)求的是小于等于a的数字个数

#define DeBUG
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <stack>
#include <queue>
#include <string>
#include <set>
#include <sstream>
#include <map>
#include <bitset>
using namespace std ;
#define zero {0}
#define INF 2000000000
#define EPS 1e-6
typedef long long LL;
const double PI = acos(-1.0);
//#pragma comment(linker, "/STACK:102400000,102400000")
inline int sgn(double x)
{
    return fabs(x) < EPS ? 0 : (x < 0 ? -1 : 1);
}
#define maxn 100001
int c[maxn];
int lowbit(int x)
{
    return x&(-x);
}
void add(int x,int num)
{
    while(x<=maxn)
    {
        c[x]+=num;
        x+=lowbit(x);
    }
}
int sum(int x)
{
    int sum=0;
    while(x>0)
    {
        sum+=c[x];
        x-=lowbit(x);
    }
    return sum;
}
int find(int a,int k)
{
    int l=1;
    int r=maxn;
    int mid;
    int s=sum(a);
    while(l<=r)
    {
        mid=(l+r)>>1;
        if(sum(mid)-s<k)
            l=mid+1;
        else
            r=mid-1;
    }
    return l;
}
int main()
{
#ifdef DeBUGs
    freopen("//home//amb//桌面//1.in", "r", stdin);
#endif
    int n;
    while (scanf("%d", &n) + 1)
    {
        memset(c,0,sizeof(c));
        int op, a, k;
        for (int i = 0; i < n; i++)
        {
            scanf("%d", &op);
            if (op == 0)
            {
                scanf("%d", &a);
                add(a, 1);
            }
            else if (op == 1)
            {
                scanf("%d", &a);
                if (sum(a) - sum(a - 1) == 0)//单点求值，是否大于1
                    printf("No Elment!\n");
                else
                    add(a, -1);
            }
            else
            {
                scanf("%d%d", &a, &k);
                if (sum(maxn) - sum(a) < k)
                    printf("Not Find!\n");
                else
                    printf("%d\n", find(a, k));
            }
        }
    }

    return 0;
}