MST(最小生成树) Test

练习链接

A-Bad Cowtractors

POJ-2377
模板题最想生成树的变形，找最大路径Kruskal的变形 排序的时候改成从大往小排就好了
AC代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int fa[2000];
struct T
{
    int r,l,data;
}b[20005];
int Find(int x)
{
    return x==fa[x]?x:fa[x]=Find(fa[x]);
}
bool cmp(const T&x,const T&y)// 排序
{
    return x.data>y.data;  //从大往小排
}
void Union(int x,int y)
{
    int fx=Find(x);
    int fy=Find(y);
    fa[fx]=fy;
}
int main()
{
    int i,j,k,n,m,xx,yy,cx=0;
    long long ans;
    while(~scanf("%d%d",&n,&m)&&n&&m)
    {
        for(i=0;i<m;i++)
            scanf("%d%d%d",&b[i].r,&b[i].l,&b[i].data);
        for(i=1;i<=n;i++)
            fa[i]=i;

        sort(b,b+m,cmp);
        ans=0;
        cx=0;
        for(i=0;i<m;i++)
        {
            if(cx==n-1)
                break;
            xx=Find(b[i].r);
            yy=Find(b[i].l);
            if(xx!=yy)
            {
                Union(xx,yy);
                ans+=b[i].data;
                cx++;
            }
        }
        if(cx==n-1)printf("%lld\n",ans);
        else
            printf("-1\n");
    }
    return 0;
}

A-Networking

POJ-1287
我耗费了半小时还是没搞懂哪里错了，一直RunTime Error。最后发同一条边存一次后就好了，只存最小的那个就解决了 T_T Kruskal的模板直接套OK！
AC代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int par[110]; 
int n,m;
int i ,j;
struct node{
	int u,v,w;
}arr[5500];
void init()
{
	for(i=1;i<=n;i++)
	par[i]=i;
}
bool cmp(node a,node b)
{
	return a.w<b.w;
}
int find(int x)
{
	return (par[x]==x)?x:par[x]=find(par[x]);
}
int unite(int x,int y)
{
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy)
	{
		par[fy]=fx;
		return 1;
	}
	return 0;
}
int main()
{
	while(scanf("%d",&n),n)
	{
		int ant=0;
		int sum=0;
		if(n==0)
		break;
		scanf("%d",&m);
		for(i=1;i<=m;i++)
		{
			scanf("%d%d%d",&arr[i].u,&arr[i].v,&arr[i].w);
		}
		sort(arr+1,arr+m+1,cmp);
		init();
		for(i=1;i<=m;i++)
		{
		 if(unite(arr[i].u,arr[i].v))
		 {
		 	ant++;
		 	sum+=arr[i].w;
		 }	
		 if(ant==n-1)
		 break;
		}
		printf("%d\n",sum);
	}
	return 0;
}

C-畅通工程再续

HDU-1875
模板题只要判断连个岛之间，是否能构建桥，能将距离数据存入，不能则跳过存下一组数据。用两点之间的距离求出他们要建桥的距离。OK
AC代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#define maxn 105
#define INF 0x3f3f3f3f
typedef long long ll;
 
using namespace std;
 
int fa[maxn];
int tol;
struct Edge
{
	int u, v;
	double w;
}edge[maxn*maxn];
struct Node
{
	int x, y;
}node[maxn];
void add(int u, int v, double w)
{
	edge[tol].u = u;
	edge[tol].v = v;
	edge[tol++].w = w;
}
int Find(int x)
{
	if (fa[x] == -1) return x;
	return fa[x] = Find(fa[x]);
}
bool cmp(Edge a, Edge b)
{
	return a.w < b.w;
}
double Kruskal(int n)
{
	sort(edge, edge + tol, cmp);
	double ans = 0;
	int cnt = 0;
	for (int i = 0; i < tol; i++) {
		int u = edge[i].u, v = edge[i].v;
		double w = edge[i].w;
		int f1 = Find(u), f2 = Find(v);
		if (f1 != f2) {
			fa[f1] = f2;
			ans += w;
			cnt++;
		}
		if (cnt == n - 1) break;
	}
	if (cnt != n - 1) return -1;
	return ans;
}
void Init()
{
	tol = 0;
	memset(fa, -1, sizeof(fa));
}
int main()
{
	int n, u, v, w, f, m, T;
 
	scanf("%d", &T);
	while (T--) {
	    Init();
		scanf("%d", &n);
		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++) {
				int dis = (node[i].x - node[j].x)*(node[i].x - node[j].x) + (node[i].y - node[j].y)*(node[i].y - node[j].y);
				double d = sqrt((double)dis);
				if (dis >= 100 && dis <= 1000000)
					add(i, j, d);
			}
		}
		double ans = Kruskal(n);
		if (ans == -1) printf("oh!\n");
		else printf("%.1f\n", ans * 100.0);
	}
	return 0;
}

D-Agri-Net

POJ-1258
修路，将每个村庄都连在一起，用最少的花费修路。模板题。

#include <iostream>
#include<algorithm>
using namespace std;
int mapn[110][110];
int n;
int k;
struct dd {
    int x,y,w;
} p[5500];
int fa[5500];
int cmp(dd o1,dd o2)
{
    return o1.w<o2.w;
}
void init()
{
    for(int i=1; i<k; i++)
        fa[i]=i;
}
int find(int x)
{
    return (fa[x]==x)?x:fa[x]=find(fa[x]);
}
int join(int x,int y)
{
    int dx=find(x);
    int dy=find(y);
    if(dx!=dy) {
        fa[dy]=dx;
        return 1;
    }
    return 0;
}
int main()
{
    while(cin>>n) {

        for(int i=1; i<=n; i++)
            for(int j=1; j<=n; j++)
                cin>>mapn[i][j];
        k=1;
        for(int i=1; i<=n; i++)
            for(int j=i+1; j<=n; j++) {
                if(mapn[i][j]>0&&mapn[i][j]<=100000) {
                    p[k].x=i;
                    p[k].y=j;
                    p[k].w=mapn[i][j];
                    k++;
                }
            }
        init();
        sort(p+1,p+k,cmp);
//    for(int i=1; i<k; i++)
//        cout<<p[i].x<<" "<<p[i].y<<" "<<p[i].w<<endl;
        int ans=0;
        for(int i=1; i<k; i++)
            if(join(p[i].x,p[i].y))
                ans+=p[i].w;
        cout<<ans<<endl;
    }
    return 0;
}

E-Building a Space Station

POJ-2031
题目高大上一点，修建空间通道，但是有些空间是重叠的，重叠的两个空间站可以认为之间距离为零，所以：要判断两个空间站是否重叠，把空间站看成一个球。球心与球心的距离，和两个半径和之间的差值。剩下就是模板。

#include <iostream>
#include <algorithm>
#include <cmath>
#include <stdio.h>
using namespace std;
int fa[9999];
int n;
int k;
struct dd {
    double x,y,z,r;
} p[9999];
struct bb {
    int x,y;
    double dis;
} T[9999];
double cmp(bb b1,bb b2)
{
    return b1.dis<b2.dis;
}
void init()
{
    for(int i=1; i<k; i++)
        fa[i]=i;
}
int find(int x)
{
    return (fa[x]==x)?x:fa[x]=find(fa[x]);
}
int join(int x,int y)
{
    int dx=find(x);
    int dy=find(y);
    if(dx!=dy) {
        fa[dy]=dx;
        return 1;
    }
    return 0;

}
int main()
{
    while(cin>>n) {
        if(n==0)break;
        for(int i=1; i<=n; i++)
            cin>>p[i].x>>p[i].y>>p[i].z>>p[i].r;
        k=1;
        for(int i=1; i<=n; i++) {
            for(int j=i+1; j<=n; j++) {
                double dis_way=sqrt((p[i].x-p[j].x)*(p[i].x-p[j].x)+
                                    (p[i].y-p[j].y)*(p[i].y-p[j].y)
                                    +(p[i].z-p[j].z)*(p[i].z-p[j].z));
                if((dis_way-p[i].r-p[j].r)>0) {
                    T[k].x=i,T[k].y=j;
                    T[k].dis=(dis_way-p[i].r-p[j].r);
                    k++;
                } else {
                    T[k].x=i,T[k].y=j;
                    T[k].dis=0.0;
                    k++;
                }
            }
        }
        sort(T+1,T+k,cmp);
        init();
        double ans=0;
        for(int i=1; i<k; i++)
            if(join(T[i].x,T[i].y))
                ans+=T[i].dis;
        printf("%.3lf\n",ans);
    }
    return 0;