hdu 2852 KiKi's K-Number （线段树，求大于x的第k小数）

 http://acm.hdu.edu.cn/showproblem.php?pid=2852

（1）求比x大的第k个最小值。问题可以分解为两个子问题：

　　　　1）求不大于x 的数的个数t。（本质为求和）

           2）求树里头的第t+y个数的值。（本质为定位）

（2）考虑过直接计算出结果：

search(y, x, n, 1);

       即在区间(x, n)中寻找第y个数。

       这种思想是可行的，但不适合已经构建的这棵树。因为已有的线段树只能从区间(1, 某数)开始查询，区间（x, n）与sum[]数组的描述方式对不上号。

　　 （原有的sum[]数组中可能不含描述区间（x, n）的sum[i]，并且一般情况下就是不包含对该区间的该描述）。

       正确的分段式处理如下：

x=add(1, x, 1, n, 1);
y+=x;
query(y, 1, n, 1);

 

具体代码：

View Code

#include<stdio.h>
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
const int maxn=101000;
int n=maxn, m;
int sum[maxn<<2];
void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void build(int l, int r, int rt)
{
    sum[rt]=0;
    if(l==r)
    {
        return ;
    }
    int m=l+r>>1;
    build(lson);
    build(rson);
}
void inserts(int val, int l, int r, int rt)
{
    if(l==r)
    {
        sum[rt]++;
        return ;
    }
    int m=l+r>>1;
    if(val<=m) inserts(val, lson);
    else inserts(val, rson);
    pushup(rt);
}
void deletes(int val, int l, int r, int rt)
{
    if(l==r)
    {
        if(sum[rt]==0) printf("No Elment!\n");
        else sum[rt]--;
        return ;
    }
    int m=l+r>>1;
    if(val<=m) deletes(val, lson);
    else deletes(val, rson);
    pushup(rt);
}
int add(int L, int R, int l, int r, int rt)
{
    if(L<=l&&r<=R)
    {
        return sum[rt];
    }
    int m=l+r>>1;
    int ret=0;
    if(L<=m) ret+=add(L, R, lson);
    if(R>m) ret+=add(L, R, rson);
    return ret;
}
void query(int p, int l, int r, int rt)
{
    if(l==r)
    {
        printf("%d\n", l);
        return ;
    }
    int m=l+r>>1;
    if(p<=sum[rt<<1]) query(p, lson);
    else query(p-sum[rt<<1], rson);
}
int main()
{
    while(scanf("%d", &m)!=EOF)
    {
        build(1, n, 1);
        for(int i=1;i<=m;i++)
        {
            int a, x, y;
            scanf("%d", &a);
            if(a==0)
            {
                scanf("%d", &x);
                inserts(x, 1, n, 1);
            }
            else if(a==1)
            {
                scanf("%d", &x);
                deletes(x, 1, n, 1);
            }
            else
            {
                scanf("%d%d", &x, &y);
                x=add(1, x, 1, n, 1);
                y+=x;
                if(y>sum[1]) printf("Not Find!\n");
                else query(y, 1, n, 1);
            }
        }
    }
    return 0;
}

 

转载于:https://www.cnblogs.com/tim11/archive/2012/08/26/2657298.html