HDU4006 求集合第K大数（线段树+离散化）

来源：互联网发布：线切割hl绘图编程软件编辑：程序博客网时间：2024/06/07 00:31

题意：

输入 I 表示在集合中插入一个数，Q表示询问当前集合中第k大数

思路：

1.树状数组加二分，可是这个算法是n(logn)^2的，对于1e6肯定TLE，

但是树状数组有一个求区间第k大的黑科技，可以自己搜一下。

2.由于本蒟篛希望能练练线段树，于是本题我用线段树

真正的题解！

首先嘛，我不确定加入集合的数的大小范围（题目似乎没给？？）

于是，果断先把修改和询问先离线下来再离散一波！

接着，在线段树上单点修改，每次将离散后的序号传进去

最重要的来了，要在线段树上二分！

如果一个节点的右儿子对应的区间和大于k那么继续向右儿子走，否则向左儿子走。

二分详细步骤在代码中

/*最怕一生碌碌无为，还说平凡难能可贵。*/#include<bits/stdc++.h>#define INF 1001000using namespace std;struct data{int l,r,sum;}tr[INF*4];int n,k,linum;int pos[INF],lisan[INF];char ch[INF];inline void R(int &x){x=0;char c=getchar();int p=1;while(c<'0'||c>'9'){if(c=='-')p=-1;c=getchar();}while(c>='0'&&c<='9'){x=(x<<1)+(x<<3)+(c^'0');c=getchar();}x*=p;}void update(int k){tr[k].sum=tr[k<<1].sum+tr[k<<1|1].sum;}void build(int k,int s,int t){tr[k].l=s;tr[k].r=t;if(s==t){tr[k].sum=0;return;}int mid=s+t>>1;build(k<<1,s,mid);build(k<<1|1,mid+1,t);update(k);}void modify(int k,int poss){if(tr[k].l==tr[k].r){tr[k].sum++;return;}int mid=tr[k].l+tr[k].r>>1;if(poss<=mid)modify(k<<1,poss);else modify(k<<1|1,poss);update(k);}int query(int k,int q){if(tr[k].l==tr[k].r)return lisan[tr[k].l];if(tr[k<<1|1].sum>=q)return query(k<<1|1,q);else return query(k<<1,q-tr[k<<1|1].sum);//"q-tr[k<<1|1].sum"是因为如果要继续访问左儿子，那么原来的第q大数，在左儿子中应该是求第 （q - 右儿子中统计的数的总和 ） }int getlisanpos(int x){return lower_bound(lisan+1,lisan+linum+1,x)-lisan;}int main(){//freopen("in.in","r",stdin);while(cin>>n>>k){int cnt=0;for(int i=1;i<=n;i++){char s[2];scanf("%s",s);ch[i]=s[0];if(ch[i]=='I'){R(pos[i]);lisan[++cnt]=pos[i];}}sort(lisan+1,lisan+cnt+1);linum=unique(lisan+1,lisan+cnt+1)-lisan-1;build(1,1,n);for(int i=1;i<=n;i++){if(ch[i]=='I')modify(1,getlisanpos(pos[i]));elsecout<<query(1,k)<<endl;}}}

1 0