首页 > 代码库 > hdu5107 线段树

hdu5107 线段树

2024-08-03 21:06:51   216人阅读

hdu 5107 这题说的是给了一个二维的 平面， 平面内有30000个点每个点都有自己的高度，然后又30000次的查询，每次查询给的是(X,Y,K), 要求出set(x,y){x,y|x<=X&&y<=Y } 的所有点，中第K 大的数是多少，存在输出该高度，允许重复； 否者输出-1，。

  本来想用主席树做，发现，找前节点还是比较麻烦的。再加上k<=10 还是比较小的， 然后我们按照y设置为线段树的页节点，然后发现每个页节点最多存10种高度，然后他们的父节点只需要合并孩子的两个长度为10的高度数组， 我们将所有的点包括询问的点进行离散化，按x按y从小到大排序， 离散化后查询在数组中， 保证了x一定小于等于当前要插入或者是查询的点，这样就可以减少对于x的处理，然后接下来就是让该点的y找到相应的叶节点，然后用该建筑物的h更新叶节点。 如果是查询就查寻（1，IDX（P[i].y））

#include <algorithm>#include <cstdio>#include <string.h>#include <vector>using namespace std;const int maxn=30005;typedef int ll;struct point{    ll x,h[11];}AN;struct build{    ll x,y,h;    ll id,op;    bool operator < ( const build A ) const {         return x<A.x||(x==A.x&&y<A.y)||(x==A.x&&y==A.y&&op<A.op)||(x==A.x&&y==A.y&&op==A.op&&id<A.id);    }}P[maxn*2];ll Y[maxn*2];ll ans[maxn];ll Va,loc,cL,cR,tim;struct Itree{   point V[maxn*4*2];   void build(int o,int L, int R){        V[o].x=0;        if(L==R) return ;        int mid =(L+R)/2;        build(o*2, L, mid );        build(o*2+1, mid+1, R);   }   void inser(int o ){       int i=0;       for( i=0; i<V[o].x; i++){         if(Va<V[o].h[i]) break;       }       if(i==10) return;       for(int j = V[o].x; j>i; --j)         V[o].h[j]=V[o].h[j-1];       V[o].h[i]=Va;       V[o].x=min(V[o].x+1,10);   }   point maintain( point o1, point o2){      point ans;       int i,j,k;       i=j=k=0;       while( (i<o1.x||j<o2.x )&&( k<10) ){             if(j>=o2.x ||( i<o1.x && o1.h[i]<o2.h[j]  ) )                  ans.h[k++] = o1.h[i++];             else ans.h[k++] = o2.h[j++];       }       ans.x=k;       return ans;   }   void update(int o, int L, int R){        if(L==R){          inser(o); return ;        }        int mid = (L+R)>>1;        if(loc<=mid) update(o*2,L, mid);        else update(o*2+1,mid+1,R);        V[o]=maintain(V[o*2], V[o*2+1]);   }   void query(int o, int L, int R){         if(cL<=L && R<= cR){            if(tim==0){                 tim=1;                 AN=V[o];            }else {                AN=maintain(AN,V[o]);            }            return ;         }         int mid = (L+R)>>1;         if(cL<=mid) query(o*2,L,mid);         if(cR>mid) query(o*2+1, mid+1, R);   }}T;int main(){   int n,m;   while(scanf("%d%d",&n,&m)==2){        for(int i=0; i<n; ++i){            scanf("%d%d%d",&P[i].x,&P[i].y,&P[i].h);             P[i].id=i;            P[i].op=0;            Y[i]=P[i].y;        }        for(int i=0; i<m; ++i){             scanf("%d%d%d",&P[i+n].x,&P[i+n].y,&P[i+n].h);             P[i+n].id=n+i;             P[i+n].op=1;             Y[i+n]=P[i+n].y;        }        sort(Y,Y+n+m);        int Ynum = unique(Y,Y+n+m)-Y;        T.build(1,1,Ynum);        sort(P,P+n+m);        for(int i=0; i<n+m; ++i){             if(P[i].op==0){                 loc = lower_bound(Y,Y+Ynum,P[i].y)-Y+1;                  Va= P[i].h;                  T.update(1, 1, Ynum);             }else{                 tim=0;                 cR = lower_bound(Y,Y+Ynum,P[i].y)-Y+1;                 cL=1;                 T.query(1,1,Ynum);                 if(AN.x>=P[i].h){                     ans[P[i].id-n]=AN.h[ P[i].h-1 ];                 } else{                     ans[P[i].id-n]=-1;                 }             }        }        for(int i=0; i<m; ++i)             printf("%d\n",ans[i]);   }   return 0;}
View Code

 

hdu5107 线段树


联系
我们