BZOJ3053: The Closest M Points

题解：

我们可以事先在堆里放入插入m个inf然后不断的比较当前值与堆首元素的大小，如果小于的话进入。

估计函数也可以随便写写。。。

query的时候貌似不用保留dir。。。

return 0写在 while 里面我也是醉了。。。

不用开long long为什么大家都开了。。。

k-d tree助我进第一版

代码：

 #include<cstdio>
 #include<cstdlib>
 #include<cmath>
 #include<cstring>
 #include<algorithm>
 #include<iostream>
 #include<vector>
 #include<map>
 #include<set>
 #include<queue>
 #include<string>
 #define inf 2100000000
 #define maxn 100000+5
 #define maxm 500+100
 #define eps 1e-10
 #define ll long long
 #define sqr(x) (x)*(x)
 #define pa pair<int,int>
 #define for0(i,n) for(int i=0;i<=(n);i++)
 #define for1(i,n) for(int i=1;i<=(n);i++)
 #define for2(i,x,y) for(int i=(x);i<=(y);i++)
 #define for3(i,x,y) for(int i=(x);i>=(y);i--)
 #define for4(i,x) for(int i=head[x],y;i;i=e[i].next)
 #define mod 1000000007
 using namespace std;
 inline int read()
 {
     int x=,f=;char ch=getchar();
     while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
     while(ch>=''&&ch<=''){x=*x+ch-'';ch=getchar();}
     return x*f;
 }
 int n,m,rt,cur,ans[];
 struct rec
 {
     int d[],mi[],mx[],l,r;
     int& operator[](int x){return d[x];}
 }p[maxn],now,t[maxn];
 priority_queue<pa>q;
 bool operator <(rec a,rec b){return a[cur]<b[cur];}
 inline void pushup(int k)
 {
     int l=t[k].l,r=t[k].r;
     for0(i,m-)
     {
         t[k].mi[i]=min(t[k].mi[i],min(t[l].mi[i],t[r].mi[i]));
         t[k].mx[i]=max(t[k].mx[i],max(t[l].mx[i],t[r].mx[i]));
     }
 }
 int build(int l,int r,int dir)
 {
     int mid=(l+r)>>;
     cur=dir;
     nth_element(p+l,p+mid,p+r+);
     t[mid]=p[mid];
     for0(i,m-)t[mid].mx[i]=t[mid].mi[i]=t[mid][i];
     t[mid].l=l<mid?build(l,mid-,(dir+)%m):;
     t[mid].r=mid<r?build(mid+,r,(dir+)%m):;
     pushup(mid);
     return mid;
 }
 inline int dis(rec a,rec b)
 {
     int ret=;
     for0(i,m-)ret+=sqr(a[i]-b[i]);
     return ret;
 }
 inline int calc(int k)
 {
     if(!k)return inf;
     int ret=;
     for0(i,m-)if(now[i]<t[k].mi[i])ret+=sqr(t[k].mi[i]-now[i]);
     for0(i,m-)if(now[i]>t[k].mx[i])ret+=sqr(now[i]-t[k].mx[i]);
     return ret;
 }
 void query(int k)
 {
     int dl=calc(t[k].l),dr=calc(t[k].r),d0=dis(t[k],now);
     if(d0<q.top().first){q.pop();q.push(pa(d0,k));}
     if(dl<dr)
     {
         if(dl<q.top().first)query(t[k].l);
         if(dr<q.top().first)query(t[k].r);
     }else
     {
         if(dr<q.top().first)query(t[k].r);
         if(dl<q.top().first)query(t[k].l);
     }
 }
 int main()
 {
     freopen("input.in","r",stdin);
     freopen("output.txt","w",stdout);
     for0(i,)t[].mi[i]=inf,t[].mx[i]=-inf;
     while(scanf("%d %d",&n,&m)!=EOF)
     {
     for1(i,n)for0(j,m-)p[i][j]=read();
     rt=build(,n,);
     int qq=read();
     while(qq--)
     {
         for0(i,m-)now[i]=read();n=read();
         printf("the closest %d points are:\n",n);
         for1(i,n)q.push(pa(inf,));
         query(rt);
         for1(i,n)ans[i]=q.top().second,q.pop();
         for3(i,n,){for0(j,m-){if(j)printf(" ");printf("%d",t[ans[i]][j]);}printf("\n");}
     }
     }
     return ;
 }