poj 1751 highways

用prim能过，用ku...超时

ku...

#include<iostream>
using namespace std;

struct town{
       int x;
       int y;       
}town[800];

struct dist{
       int from;
       int to;
       int dis;
       int vis;
}dist[320000];

int cmp(const void *a, const void *b)
{
    return (*(struct dist *)a).dis > (*(struct dist *)b).dis ? 1 : -1;
}

int fa[800];
int find(int x)
{
    if(x == fa[x])
         return x;
    fa[x] = find(fa[x]);
    return fa[x];
}

int main()
{
    int N, M, i;
    cin>>N;
    for(i=0;i<N;i++)
    {
          cin>>town[i].x>>town[i].y;
    }
    
    cin>>M;
    for(i=0;i<M;i++)
    {
          cin>>dist[i].from>>dist[i].to;
          dist[i].dis = 0;
          dist[i].vis = 0;
    }
    
       int j, k;
       i = M;
        for(j=0;j<N;j++)
        {
             for(k=j+1;k<N;k++)
             {
                  
     
                  dist[i].vis = 1;
                  dist[i].from = j+1;
                  dist[i].to = k + 1;
                  dist[i].dis = (town[j].x-town[k].x)*(town[j].x-town[k].x) + (town[j].y-town[k].y)*(town[j].y-town[k].y);
                  i++;
             }
        }
    
    
    int num = M+N*(N-1)/2;
    qsort(dist, num, sizeof(dist[0]), cmp);
    
    
    for(i=1;i<=N;i++)
        fa[i]=i;
    
    int count = 0;
    for(i=0;i<num;i++)
    {
        if(find(dist[i].from) != find(dist[i].to))
        {
            fa[find(dist[i].from)] = find(dist[i].to);
            count++;
            if(dist[i].vis)
                   cout<<dist[i].from<<" "<<dist[i].to<<endl;
            
            if(count == N-1)
                break; 
        }
    }
    
    system("pause");
    return 0;
}

prim

#include<iostream>
#include<climits>
#include<cstring>
using namespace std;

struct node{
       int x, y;
}node[755];

int maxEdge=INT_MAX;
int flag[755];
int pre[755];
int dist[755];

int map[755][755];
int main()
{
    int N, M;
    cin>>N;
    for(int i=1;i<=N;i++)
    {
            cin>>node[i].x>>node[i].y;
    }
    memset(map, 0, sizeof(map));
    for(int i=1;i<=N;i++)
    {
            for(int j=1;j<=N;j++)
            {
                    map[i][j]=(node[i].x-node[j].x)*(node[i].x-node[j].x)+(node[i].y-node[j].y)*(node[i].y-node[j].y);
            }
    }
    for(int i=0;i<=N;i++)
    {
            flag[i]=0;
            dist[i]=maxEdge;
            pre[i]=1;
    }
    
    cin>>M;
    for(int i=1;i<=M;i++)
    {
         int a, b;
         cin>>a>>b;
         pre[b] = a;
         map[a][b]=0;
         map[b][a]=0;   
    }
    
   
    flag[1]=1;
    dist[1]=maxEdge;  
    for(int j=2;j<=N;j++)
    {
         dist[j]=map[1][j];
    }
      
    int next,p=1; 
    for(int i=1;i<N;i++)
    {  
        int min=maxEdge; 
        
        for(int j=1;j<=N;j++)
        {
            if(dist[j]<min)
            {
                min=dist[j];
                next = j;
            }                  
        }
         p=pre[next];
         flag[next]=1;
         dist[next]=maxEdge;
         pre[next]=p;
         if(map[pre[next]][next]  != 0)
         {
              cout<<pre[next]<<" "<<next<<endl;
         }
        for(int j=1;j<=N;j++)
        {
                if((map[next][j] < dist[j])&& flag[j]==0)
                {
                    dist[j]=map[next][j];
                    pre[j]=next;
                }
        }           
           
    }
    system("pause");
    return 0;
}