POJ T1875 畅通工程再续

                              POJ T1875 畅通工程再续

---

题解：

  前面先输入整个矩阵，再计算每个点到其他点的距离。但要看清楚题目给的条件：2个小岛之间的距离不能小于10米，也不能大于1000米。当然，为了节省资金，只要求实现任意2个小岛之间有路通即可。其中桥的价格为 100元/米。

  在选边的时候要记得判断，并且Kruskal写法的话是遍历父节点数组f，看是否只有一个根节点来判断是否连通。而Prim是将不符合条件的边赋值为INF，如果在Prim过程中minn==INF，则不连通。

 

Kruskal写法

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
#define maxn 105
#define INF 0x3f3f3f3f
using namespace std;

int n,cnt,cnte;
double sum;
int f[maxn];
 

struct Edge{
	int u,v;
	double c;
}edge[maxn*maxn];

struct Node{
	int x,y;
}node[maxn];

bool cmp(Edge a,Edge b){
	return a.c < b.c;
}

void init(){
	sum = 0;
	cnt = cnte = 0;
	for(int i = 1; i <= n ; i++)
	    f[i] = i;
}

double solve(int i,int j){
	double a = (node[i].x - node[j].x)*(node[i].x - node[j].x);
	double b = (node[i].y - node[j].y)*(node[i].y - node[j].y);
	return sqrt(a+b);
}

int getf(int x){
	return f[x] == x ? x : f[x] = getf(f[x]);
}

bool merge(int u,int v){
	int rootu = getf(u);
	int rootv = getf(v);
	if(rootu != rootv){
		f[rootv] = rootu;
		return true;
	}
	return false;
}

void Kruskal(){
	sort(edge,edge+cnt,cmp);
	for(int i = 0 ; i < cnt; i++){
		if(edge[i].c >=10 && edge[i].c <= 1000 && merge(edge[i].u,edge[i].v)){
			sum += edge[i].c;
			cnte++;
		}
		if(cnte == n-1) break;
	}
}

int main(){
	int t;
	scanf("%d",&t);
	while(t--){
	    scanf("%d",&n);
	    init();
	    for(int i = 1; i <= n ;i++)
	        scanf("%d%d",&node[i].x,&node[i].y);
	    for(int i = 1; i <= n ;i++)
	        for(int j = i+1; j <= n ;j++){
	        	edge[cnt].u = i;
	        	edge[cnt].v = j;
	        	edge[cnt++].c = solve(i,j);
	        }
	    Kruskal();
	    int cntf = 0;
		for(int i = 1; i <= n; i++)
	        if(f[i] == i) cntf++;
	    if(cntf != 1)
		    puts("oh!");
		else 
		    printf("%.1lf\n",sum*100);
	}
	return 0;
}

---

Prim写法

#include<cstdio>
#include<cstring>
#include<cmath>
#define INF 0xfffffff

double map[125][125],low[125],sum;
int x[125],y[125];   //map二维数组存图，low记录每2个点间最小权值，vis标记某点是否已访问
int vis[125];
int n;

double dlen(int i,int j)
{
	return sqrt( 1.0*((x[i]-x[j])*(x[i]-x[j])+(y[i]-y[j])*(y[i]-y[j])) );
}

void prim()
{
	double min;
	int i,j,pos;
	sum = 0;   
	memset(vis,0,sizeof(vis));
	vis[1]=1;	pos=1;   	//从某点开始，分别标记vis和记录该点pos
	for(i=1;i<=n;++i)   //第一次给low数组赋值 map的第一行 
		low[i]=map[pos][i];

	for(i=1;i<n;++i)   //再运行n-1次,一次找一个最小 
	{
		min=INF;
		for(j=1;j<=n;++j)
		{
			if(vis[j]==0&&low[j]<min)
			{
				min=low[j];
				pos=j;
			}
		}
		
		if(min==INF)  //不能连通  找不到符合条件的 
		{
			printf("oh!\n");
			return ;
		}
		
		vis[pos]=1;    //标记该点已访问 
		sum+=min;   //最小权值累加
		for(j=1;j<=n;++j)    //更新权值low 把 map的 pos 行中比对应的 low 小的赋给low 
			if(vis[j]==0&&low[j]>map[pos][j])
				low[j]=map[pos][j];
	}
}

int main()
{
	int T,i,j;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&n);
		for(i=1;i<=n;++i)
			scanf("%d%d",&x[i],&y[i]);
		memset(map,0,sizeof(map));
		for(i=1;i<=n;++i)
		{
			for(j=1;j<=n;++j)
			{
				map[i][j]=map[j][i]=dlen(i,j);
				if(map[i][j]<10||map[i][j]>1000)  //不符合条件 将权值设为无穷大，不会用再到
					map[i][j]=map[j][i]=INF;
			}
		}
		
		prim();
		printf("%.1lf\n",sum*100);
	}
	return 0;
}