poj 1258 Agri-Net

这是一道最小生成树的问题。题目输入的二维数组就是用prim算法展开的邻接表所以我选择了prim算法。

#include<stdio.h>
#define MAX 10000000
#define maxn 105 
int map[maxn][maxn];
int prim(int n)
{
	int i,j,k;
	int low[maxn];
	int pos[maxn];
	int min,sum=0;
	for(i=2;i<=n;i++)
	{
		low[i]=map[1][i];
		pos[i]=1;
	}
	pos[1]=0;
	for(i=2;i<=n;i++)
	{
		min=MAX;
		k=0;
		for(j=2;j<=n;j++)
		{
			if(low[j]<min&&low[j]!=0)
			{
				min=low[j];
				k=j;
			}
		}
		sum+=min;
		low[k]=0;
		for(j=2;j<=n;j++)
		{
			if(map[k][j]<low[j])
			{
				low[j]=map[k][j];
				pos[j]=k;
			}
		}
	}
	return sum;
}
int main()
{
	int n,i,j;
	while(~scanf("%d",&n))
	{
		for(i=1;i<=n;i++)
		{
			for(j=1;j<=n;j++)
				map[i][j]=MAX;
		}
		for(i=1;i<=n;i++)
		{
			for(j=1;j<=n;j++)
				scanf("%d",&map[i][j]);
		}
		printf("%d\n",prim(n));
	}
	return 0;
}

 

然后，我在网上找到一个用krusal 算法写的。做一个比较吧。

#include<iostream>
using namespace std; 
int map[101][101]; 
int father[10001]; 
int find(int x) 
{
    int r=x; 
    while(r!=father[r]) 
        r=father[r];
    int temp; 
    while(r!=x) 
    {
        temp=father[x];
        father[x]=r;
        x=temp;
    }
    return r; 
}
void merge(int x,int y) 
{
    father[x]=y;
}
int main() 
{               
    int n; 
    while(cin>>n) 
    {
        int i,j,k; 
        //memset(visited,0,sizeof(visited));
 for(i=0;i<=n*n;i++) 
            father[i]=i;
        memset(map,0,sizeof(map)); 
        for(i=0;i<n;i++) 
            for(j=0;j<n;j++) 
                scanf("%d",&map[i][j]); 
        int x,y; 
        int sum=0; 
        int fx,fy; 
        for(i=1;i<n;i++) 
        {
            int min=1000000; 
            for(j=0;j<n;j++) 
                for(k=j;k<n;k++) 
                {
                
                    if(map[j][k]<min) 
                    {       
                        x=find(j);
                        y=find(k);
                    //    cout<<x<<"  "<<y<<endl;
 if(x!=y) 
                        {   
                        //    cout<<"x:"<<x<<" "<<"y:"<<y<<endl;
 min=map[j][k]; 
                            fx=x;
                            fy=y;
                            //merge(x,y);
 } 
                    }
                }
                merge(fx,fy);
            //    cout<<"min:"<<min<<endl;
 sum+=min; 
        }
        cout<<sum<<endl;
        
    }
    return 0; 
}