划分树:
原理: 划分树的建树过程:
基本就是模拟快排过程,取一个已经排过序的区间中值,然后把小于中值的点放左边,大于的放右边, 并记录划分的每个区间中放在左边的数的个数,另外将放的结果记录在下一个tree[ ][ ]中,然后左右建树。
划分树的查找过程:
通过计算所要查找区间放在左边的数的个数分两种情况,然后寻找下一个查找的子区间(具体看代码)。
ps: 如果要求区间和,则需要另外加sum[ ] [ ]记录每一个区间的和。
HDU 2665 Kth number(划分树入门题,纯套模板)
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<string>
#include<map>
#include<set>
#include<queue>
#include<vector>
#include<stack>
#include<cstdlib>
using namespace std;
const int maxn=100010;
int A[maxn];
int tree[25][maxn];
int sum[25][maxn];
void build(int c,int l,int r)
{
if(l==r) return ;
int mid=(l+r)>>1;
int same=mid+1-l;
for(int i=l;i<=r;i++){
if(tree[c][i]<A[mid]) same--;
}
int sl=l;
int sr=mid+1;
for(int i=l;i<=r;i++){
if(tree[c][i]<A[mid]){
tree[c+1][sl++]=tree[c][i];
}
else if(tree[c][i]==A[mid]&&same>0){
tree[c+1][sl++]=tree[c][i];
same--;
}
else{
tree[c+1][sr++]=tree[c][i];
}
sum[c][i]=sum[c][l-1]+sl-l;
}
build(c+1,l,mid);
build(c+1,mid+1,r);
}
int query(int c,int L,int R,int l,int r,int k){
if(l==r) return tree[c][l];
int mid=(L+R)>>1;
int cnt=sum[c][r]-sum[c][l-1];
if(cnt>=k){
int newl=L+sum[c][l-1]-sum[c][L-1];
int newr=newl+cnt-1;
return query(c+1,L,mid,newl,newr,k);
}
else{
int newr=r+sum[c][R]-sum[c][r];
int newl=newr-(r-l-cnt);
return query(c+1,mid+1,R,newl,newr,k-cnt);
}
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("in.cpp","r",stdin);
#endif // ONLINE_JUDGE
int n,m;
int T;
cin>>T;
while(T--){
scanf("%d%d",&n,&m);
memset(tree,0,sizeof tree);
for(int i=1;i<=n;i++){
scanf("%d",&A[i]);
tree[0][i]=A[i];
}
sort(A+1,A+n+1);
build(0,1,n);
int s,t,k;
while(m--){
scanf("%d%d%d",&s,&t,&k);
printf("%d\n",query(0,1,n,s,t,k));
}
}
return 0;
}
hdu 3473
涉及到区间和,找中位数。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<string>
#define LL unsigned long long
using namespace std;
const int maxn=100010;
LL ans;
LL sum[20][maxn];
int A[maxn];
int toleft[20][maxn];
int tree[20][maxn];
void build(int c,int l,int r){
if(l==r) {
sum[c][l]=sum[c][l-1]+tree[c][l];
return;
}
int mid=(l+r)>>1;
int same=mid+1-l;
for(int i=l;i<=r;i++){
if(tree[c][i]<A[mid])
same--;
sum[c][i]+=sum[c][i-1]+tree[c][i];
}
int sl=l;
int sr=mid+1;
for(int i=l;i<=r;i++){
if(tree[c][i]<A[mid])
tree[c+1][sl++]=tree[c][i];
else if(tree[c][i]==A[mid]&&same>0){
tree[c+1][sl++]=tree[c][i];
same--;
}
else{
tree[c+1][sr++]=tree[c][i];
}
toleft[c][i]=toleft[c][l-1]+sl-l;
}
build(c+1,l,mid);
build(c+1,mid+1,r);
}
int query(int c,int L,int R,int l,int r,int k){
if(l==r) return tree[c][l];
int mid=(L+R)>>1;
int cnt=toleft[c][r]-toleft[c][l-1];
if(cnt>=k){
int ee=r-(L-1)-(toleft[c][r]-toleft[c][L-1])+mid;
int ss=l-1-(L-1)-(toleft[c][l-1]-toleft[c][L-1])+mid;
ans+=sum[c+1][ee]-sum[c+1][ss]; //注意这里的ss不需要-1;
int newl=L+toleft[c][l-1]-toleft[c][L-1];
int newr=newl+cnt-1;
return query(c+1,L,mid,newl,newr,k);
}
else {
int s=L+(toleft[c][l-1]-toleft[c][L-1]);
int e=s+cnt-1;
ans-=sum[c+1][e]-sum[c+1][s-1]; //注意此时的s-1;
int newr=r+toleft[c][R]-toleft[c][r];
int newl=newr-(r-l+1-cnt)+1;
return query(c+1,mid+1,R,newl,newr,k-cnt);
}
}
int main()
{
#ifndef ONLINE_JUDGE
freopen("in.cpp","r",stdin);
#endif // ONLINE_JUDGE
int T;
cin>>T;
int Cas=0;
while(T--){
int n;
memset(tree,0,sizeof tree);
memset(sum,0,sizeof sum);
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d",&A[i]);
tree[0][i]=A[i];
}
sort(A+1,A+1+n);
build(0,1,n);
int m,ct,ss;
scanf("%d",&m);
printf("Case #%d:\n",++Cas);
while(m--){
int s,t;
ans=0;
scanf("%d%d",&s,&t);
s++;
t++;
ss=t-s+1;
ct=query(0,1,n,s,t,ss/2+1);
if(ss%2==0) ans+=ct;
printf("%lld\n",ans);
}
cout<<endl;
}
return 0;
}
相关题型:hdu 4417 hdu 3473 hdu 2665 hdu 4215