xmu1166求第k小數 樹狀數組/線段樹

 http://acm.xmu.edu.cn/JudgeOnline/problem.php?id=1165

 

題意：有3中操作

1 a將a加入，如果a已經存在忽略該操作

2 a將a刪除，如果不存在忽略該操作

3 a查詢第a大的數，不存在輸出-1

 

分析：思想和pku2182一樣，比2182簡單。。。直接線段樹或樹狀數組+二分。。。

 

其實從這個角度而言，樹狀數組也可以求最大最小值了。。。呵呵，求最小值就是第1小值嘛。。。

 

代碼：

#include <stdio.h>#include <algorithm>#include <cmath>#include <iostream>using namespace std;const int N=100002;int a[N], sum[N], n;int query(int i){int tmp = 0;for( ; i>0; i-=i&(-i))tmp += sum[i];return tmp;}void update(int i, int v){for( ; i<N; i+=i&(-i))sum[i] += v;}int bs(int t){int l, r, mid;l=1;r=N-1;while(l<=r){mid = (l+r)/2;if(query(mid)>=t)r = mid-1;elsel = mid+1;}if(l==N)return -1;return l;}int main(){int i, j, x, y, ans;//freopen("D.in", "w", stdout);while(scanf("%d", &n)!=EOF){for(i=0; i<N; i++){sum[i] = 0;a[i] = 0;}while(n--){scanf("%d%d", &x, &y);if(x==1){if(a[y]==0){update(y, 1);a[y] = 1;}}else if(x==2){if(a[y]!=0){update(y, -1);a[y] = 0;}}else{ans = bs(y);printf("%d\n", ans);}}}return 0;}