POJ 2104/TYVJ 1253 划分树模板

 

求区间第k小值~

详细注释见代码：

 

View Code 
 1 #include <cstdio>
 2 #include <cstring>
 3 #include <iostream>
 4 #include <algorithm>
 5 #define M 100005
 6 using namespace std;
 7 int sorted[M],toleft[30][M],tree[30][M],m,n;
 8 void read()
 9 {
10     for(int i=1;i<=m;i++)
11     {
12         scanf("%d",&sorted[i]);
13         tree[0][i]=sorted[i];
14     }
15     sort(sorted+1,sorted+1+m);
16 }
17 void create(int l,int r,int d)
18 {
19     if(l==r) return;
20     int mid=(l+r)>>1;
21     int same=mid-l+1;//左子树中的数字个数 
22     for(int i=l;i<=r;i++)
23         if(tree[d][i]<sorted[mid]) same--;//此时same成为和中位数相等的位于左子树的数字个数 
24         
25     int ls=l,rs=mid+1;//分别为左右区间的起点 
26     for(int i=l;i<=r;i++)
27     {
28         int fg=0;
29         if((tree[d][i]<sorted[mid])||(tree[d][i]==sorted[mid]&&same>0))//将符合条件的数分到左子树 
30         {
31             fg=1;
32             tree[d+1][ls++]=tree[d][i];
33             if(tree[d][i]==sorted[mid]) same--;//有些与中位数相同的数分到左子树，剩下的分到右子树 
34         }
35         else tree[d+1][rs++]=tree[d][i];//分到右子树 
36         toleft[d][i]=toleft[d][i-1]+fg;//toleft[d][i]表示从l到i区间中有多少数分到了其左子树中 
37     }
38     //递归建树 
39     create(l,mid,d+1);
40     create(mid+1,r,d+1);
41 }
42 int query(int ql,int qr,int k,int l,int r,int d)
43 {
44     if(ql==qr) return tree[d][ql];
45     int mid=(l+r)>>1;
46     int x=toleft[d][ql-1]-toleft[d][l-1];//位于ql（待求区间左端点）（不包括ql）左边的放于左子树中的数字个数 
47     int y=toleft[d][qr]-toleft[d][l-1];//位于qr（待求区间右端点）（包括qr）左边的放于左子树中的数字个数
48     int ry=qr-l+1-y;//位于qr（包含qr）的在右子树中的数字个数 
49     int cnt=y-x;//位于区间[ql,qr]中的在左子树中的数字个数 
50     int rx=ql-l-x;//位于ql左边（不包含ql）的在右子树中的数字个数 
51     if(cnt>=k) return query(l+x,l+y-1,k,l,mid,d+1);
52     else return query(mid+rx+1,mid+ry,k-cnt,mid+1,r,d+1);
53 }
54 void go()
55 {
56     create(1,m,0);
57     for(int i=1,a,b,k;i<=n;i++)
58     {
59         scanf("%d%d%d",&a,&b,&k);
60         int ans=query(a,b,k,1,m,0);
61         printf("%d\n",ans);
62     }
63 }
64 int main()
65 {
66     while(scanf("%d%d",&m,&n)!=EOF)
67     {
68         read();
69         go();
70     }
71     return 0;
72 }

 

posted @ 2012-10-07 09:25  proverbs  阅读(215)  评论(0)  编辑  收藏  举报