厦门1165 第K小数 树状数组 不重复数字

1165.第K小数
Time Limit: 1000 MS         Memory Limit: 65536 K
Total Submissions: 467 (117 users)         Accepted: 117 (66 users)
[ My Solution ]

Description
给定一个空的正整数集合S，然后再给定一组操作，这组操作共有三种
1. 添加某个正整数a到集合中，若a已经在S中则不需要重复加入
2. 从集合中删除某个整数a，若a不在S中则忽略该操作
3. 查询集合中的第k小数，若不存在第k小数则输出-1,否则输出第k小数

 

Input
 第一行为正整数n(2<=n<=100000)分别表示操作的个数
 接下来n行每行为两个数字表示一个操作。第一个数字为操作的编号，1表示添加操作，2表示删除操作，3表示查询操作，
 第二个数字为要添加删除的正整数a(0<a<=100000)或要查询的第k小数k。

 

Output
对于每一个查询操作，若不存在第k小数则输出-1,否则输出第k小数

 

Sample Input
10

1 1
1 1
2 3
3 1
3 2
2 1
1 2
3 1
1 1
3 1

 

Sample Output
1
-1
2
1

 

Hint
输入输出巨大，如果您使用C/C++建议用scanf和printf输入输出。

 

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int n=100000;
int c[n+1];
inline int lowbit(int x)
{
    return x&(-x);
}
void update(int x,int val)
{
    for(int i=x;i<=n;i+=lowbit(i))
    {
        c[i]+=val;
    }
}
int getsum(int x)
{
    int cnt=0;
    for(int i=x;i>=1;i-=lowbit(i))
    {
        cnt+=c[i];
    }
    return cnt;
}
int binary_search(int l,int r,int goal)
{
    if(l<r)
    {
        int mid=(l+r)>>1;
        if(getsum(mid)==goal) return mid;
        else if(getsum(mid)>goal) return binary_search(l,mid,goal);
        else return binary_search(mid+1,r,goal);
    }
    return -1;
}
int main()
{
    int ci;scanf("%d",&ci);
    while(ci--)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        if(x==1)
        {
            if((getsum(y)-getsum(y-1))==0)
            {
                update(y,1);
            }
        }
        else if(x==2)
        {
            if((getsum(y)-getsum(y-1))==1)
            {
                update(y,-1);
            }
        }
        else
        {
            int cnt=binary_search(1,n+1,y);
            if(cnt==-1) printf("-1/n");
            else
            {
                for(;cnt>=1&&getsum(cnt)==y;cnt--);cnt++;
                printf("%d/n",cnt);
            }
        }
    }
    return 0;
}
/*
10

1 1
1 1
2 3
3 1
3 2
2 1
1 2
3 1
1 1
3 1
*/

程序员求签  http://6.kongzhidea.duapp.com/demos/kk/index.php/fun/pray