hdu 1233 还是畅通工程

还是畅通工程

Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 67160 Accepted Submission(s): 30440

Problem Description

某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。

Output

对每个测试用例，在1行里输出最小的公路总长度。

Sample Input

1 2 1

1 3 2

2 3 4

1 2 1

1 3 4

1 4 1

2 3 3

2 4 2

3 4 5

Sample Output

Hint

Huge input, scanf is recommended.

Prim

#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
int a[105][105],dis[105],vis[105];//vis标记处理过的点，dis[i]表示从起点到i点的权值
int n,ans;
void prim()
{
  ans=0;
  memset(vis,0,sizeof(vis));
  for(int i=2;i<=n;i++)//初始化
    dis[i]=a[1][i];
  dis[1]=0;
  vis[1]=1;

  for(int i=1;i<n;i++)//最后一个点不需要处理，直接加入即可，所以不要（也不能）取等
  {
    int k=0,mn=9999999;
    for(int j=1;j<=n;j++)//找出还没有被标记的点中离起点权值最小的点
    {
      if(!vis[j]&&dis[j]<mn)
      {
        mn=dis[j];
        k=j;
      }
    }
    vis[k]=1;
    ans=ans+mn;
    for(int j=1;j<=n;j++)//更新最小值，k和起点都在处理过的集合里面，更新到起点的最小值
    {
      if(!vis[j]&&dis[j]>a[k][j])
        dis[j]=a[k][j];
    }
  }

}
int main()
{
  while(~scanf("%d",&n)&&n)
  {
    int x,y,v;
    for(int i=1;i<=n*(n-1)/2;i++)
    {
      cin>>x>>y>>v;
      a[x][y]=a[y][x]=v;
    }
    prim();
    printf("%d
",ans);
  }
}

Kruskal

#include<iostream>
#include<algorithm>
using namespace std;
int p[105],r[105];
int n,ans;
struct node
{
  int x;
  int y;
  int v;
}a[100005];
bool cmp(node b,node c)
{
  return b.v<c.v;
}
int find(int x)//查找元素x的老板是谁
{
    if (x == p[x])
        return x;
    else
        return p[x] = find(p[x]);
}

void join(int x, int y)//合并两个集合
{
    int xRoot = find(x);
    int yRoot = find(y);

    if (xRoot == yRoot) //老板相同，不合并
        return;
    //cnt=cnt-1;
    if (r[xRoot] < r[yRoot]) //r[i]是元素i所在树的高度，矮树的根节点认高树的根节点做老板
        p[xRoot] = yRoot;
    else if (r[xRoot] > r[yRoot])
        p[yRoot] = xRoot;
    else
    {
        p[yRoot] = xRoot;//树高相同，做老板的树高度要加一
        r[xRoot]++;
    }
}
void kruskal()
{
  for(int i=1;i<=n;i++)//初始化根节点
    p[i]=i;
  sort(a+1,a+n*(n-1)/2+1,cmp);
  for(int i=1;i<=n*(n-1)/2;i++)
  {
    if(find(a[i].x)!=find(a[i].y))
    {
      join(a[i].x,a[i].y);
      ans=ans+a[i].v;
    }
  }
}
int main()
{
  while(~scanf("%d",&n))
  {
    ans=0;
    if(n==0)
      break;
    else
    {
      for(int i=1;i<=n*(n-1)/2;i++)
        scanf("%d %d %d",&a[i].x,&a[i].y,&a[i].v);
        //cin>>a[i].x>>a[i].y>>a[i].v;
      kruskal();
    }
    printf("%d
",ans);
    //cout<<ans<<endl;
  }
  return 0;
}