poj 1751 Highways

这题极大的摧残了我。不断的TLE WA。起先我用的是kruskal，结果TLE了n次，论证了稠密图应该用prim的。最后改用prim，但是一开始不知道怎么解决已经存在的路，最后看了别人的解题报告，说cost直接赋值为0，果然是好方法。信心满满的写了prim，然后不断的WA，去下了测试数据，还是不知道错在哪里。最后discuss中找到了答案，只有一组测试数据，EOF是不行的，哦，上帝啊。怎么能这样呢

#include<iostream>#include<cmath>#include<cstdio>using namespace std;#define max_int 99999999#define N 800struct Point{int x,y;};Point p[N];Point result[N];int s[N];int dist[N],cost[N][N];int pi[N];int choose(int n){int pos=-1;int m=max_int;for(int i=1;i<=n;i++)if(!s[i]&&dist[i]<m){pos=i;m=dist[i];}return pos;}int Distance(Point a,Point b){return (a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y);}int prim(int n){int u,i,t=0;dist[1]=0;while(1){u=choose(n);if(u==-1)break;s[u]=1;if(dist[u]>0){result[t].x=u;result[t].y=pi[u];t++;}for(i=1;i<=n;i++)if(cost[u][i]<dist[i]&&!s[i]){pi[i]=u;dist[i]=cost[u][i];}}return t;}int main(){int n,m,i,j,k,c;scanf("%d",&n);for(i=1;i<=n;i++){cost[i][i]=0;dist[i]=max_int;s[i]=0;}for(i=1;i<=n;i++)scanf("%d%d",&p[i].x,&p[i].y);for(i=1;i<=n;i++)for(j=i+1;j<=n;j++)cost[i][j]=cost[j][i]=Distance(p[i],p[j]);scanf("%d",&m);for(i=1;i<=m;i++){scanf("%d%d",&j,&k);cost[j][k]=cost[k][j]=0;}c=prim(n);for(i=0;i<c;i++)printf("%d %d/n",result[i].x,result[i].y);return 0;}