首页 > 代码库 > BZOJ3053: The Closest M Points

BZOJ3053: The Closest M Points

题解:

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

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

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

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

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

k-d tree助我进第一版

代码:

技术分享
  1 #include<cstdio>  2 #include<cstdlib>  3 #include<cmath>  4 #include<cstring>  5 #include<algorithm>  6 #include<iostream>  7 #include<vector>  8 #include<map>  9 #include<set> 10 #include<queue> 11 #include<string> 12 #define inf 2100000000 13 #define maxn 100000+5 14 #define maxm 500+100 15 #define eps 1e-10 16 #define ll long long 17 #define sqr(x) (x)*(x) 18 #define pa pair<int,int> 19 #define for0(i,n) for(int i=0;i<=(n);i++) 20 #define for1(i,n) for(int i=1;i<=(n);i++) 21 #define for2(i,x,y) for(int i=(x);i<=(y);i++) 22 #define for3(i,x,y) for(int i=(x);i>=(y);i--) 23 #define for4(i,x) for(int i=head[x],y;i;i=e[i].next) 24 #define mod 1000000007 25 using namespace std; 26 inline int read() 27 { 28     int x=0,f=1;char ch=getchar(); 29     while(ch<0||ch>9){if(ch==-)f=-1;ch=getchar();} 30     while(ch>=0&&ch<=9){x=10*x+ch-0;ch=getchar();} 31     return x*f; 32 } 33 int n,m,rt,cur,ans[10]; 34 struct rec 35 { 36     int d[5],mi[5],mx[5],l,r; 37     int& operator[](int x){return d[x];} 38 }p[maxn],now,t[maxn]; 39 priority_queue<pa>q; 40 bool operator <(rec a,rec b){return a[cur]<b[cur];} 41 inline void pushup(int k) 42 { 43     int l=t[k].l,r=t[k].r; 44     for0(i,m-1) 45     { 46         t[k].mi[i]=min(t[k].mi[i],min(t[l].mi[i],t[r].mi[i])); 47         t[k].mx[i]=max(t[k].mx[i],max(t[l].mx[i],t[r].mx[i])); 48     } 49 } 50 int build(int l,int r,int dir) 51 { 52     int mid=(l+r)>>1; 53     cur=dir; 54     nth_element(p+l,p+mid,p+r+1); 55     t[mid]=p[mid]; 56     for0(i,m-1)t[mid].mx[i]=t[mid].mi[i]=t[mid][i]; 57     t[mid].l=l<mid?build(l,mid-1,(dir+1)%m):0; 58     t[mid].r=mid<r?build(mid+1,r,(dir+1)%m):0; 59     pushup(mid); 60     return mid; 61 } 62 inline int dis(rec a,rec b) 63 { 64     int ret=0; 65     for0(i,m-1)ret+=sqr(a[i]-b[i]); 66     return ret; 67 } 68 inline int calc(int k) 69 { 70     if(!k)return inf; 71     int ret=0; 72     for0(i,m-1)if(now[i]<t[k].mi[i])ret+=sqr(t[k].mi[i]-now[i]); 73     for0(i,m-1)if(now[i]>t[k].mx[i])ret+=sqr(now[i]-t[k].mx[i]); 74     return ret; 75 } 76 void query(int k) 77 { 78     int dl=calc(t[k].l),dr=calc(t[k].r),d0=dis(t[k],now); 79     if(d0<q.top().first){q.pop();q.push(pa(d0,k));} 80     if(dl<dr) 81     { 82         if(dl<q.top().first)query(t[k].l); 83         if(dr<q.top().first)query(t[k].r); 84     }else 85     { 86         if(dr<q.top().first)query(t[k].r); 87         if(dl<q.top().first)query(t[k].l); 88     } 89 } 90 int main() 91 { 92     freopen("input.in","r",stdin); 93     freopen("output.txt","w",stdout); 94     for0(i,4)t[0].mi[i]=inf,t[0].mx[i]=-inf; 95     while(scanf("%d %d",&n,&m)!=EOF) 96     { 97     for1(i,n)for0(j,m-1)p[i][j]=read(); 98     rt=build(1,n,0); 99     int qq=read();100     while(qq--)101     {102         for0(i,m-1)now[i]=read();n=read();103         printf("the closest %d points are:\n",n);    104         for1(i,n)q.push(pa(inf,0));105         query(rt);106         for1(i,n)ans[i]=q.top().second,q.pop();107         for3(i,n,1){for0(j,m-1){if(j)printf(" ");printf("%d",t[ans[i]][j]);}printf("\n");}108     }109     }110     return 0;111 }
View Code

 

BZOJ3053: The Closest M Points