hdu1863-畅通工程(Prime & Kruskal)

畅通工程

Time Limit: 1000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 27327 Accepted Submission(s): 11956

Problem Description
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。经过调查评估，得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序，计算出全省畅通需要的最低成本。

Input
测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M ( < 100 )；随后的 N
行对应村庄间道路的成本，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间道路的成本（也是正整数）。为简单起见，村庄从1到M编号。当N为0时，全部输入结束，相应的结果不要输出。

Output
对每个测试用例，在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通，则输出“?”。

Sample Input
3 3
1 2 1
1 3 2
2 3 4
1 3
2 3 2
0 100

Sample Output
3
?

Prime

#include<cstdio>
#include<iostream>
#include<string.h>
#define INF 0x3f3f3f3f
using namespace std;
const int V=1000;
const int E=100000;
int graph[V][V];
int cost[V];  //记录目前要把第i个点加入正确联盟所需的花费
int last[V];  // 记录第i个点 透过谁加入的正确联盟
int visit[V]; // 记录是否加入正确联盟
int fin_cnt;  //记录加入正确联盟的个数
int tot_cost;  // 记录总花费

void init(int N){ //N:点数
  memset(visit,0,sizeof(visit));
  memset(last,-1,sizeof(last));

  cost[1] = 0;
  visit[1] =1;
  for(int i = 2; i <= N;i++){
    cost[i] = graph[1][i];
    if(cost[i] != INF)
      last[i]=1;
  }
  fin_cnt=1;
}

int prime(int N){
  int Min;
  int Min_idx;
  while(fin_cnt < N){
    Min = INF;
    Min_idx=-1;
    for(int i = 2; i <= N ;i++){
      if(visit[i])
        continue;
        if(cost[i] < Min)
          Min=cost[i],Min_idx=i;
    }

    if(Min_idx==-1)  break;
    visit[Min_idx] =1;
    tot_cost += cost[Min_idx];
    fin_cnt++;
    // 看看还有没有被选择的点能透过Min_idx 变近的
    for(int i=2;i<=N;i++){
      if(visit[i] ) // 被选择就跳过
        continue;
      if(graph[Min_idx][i] < cost[i] )  //有更近就更新
        cost[i]=graph[Min_idx][i],last[i]=Min_idx;
    }

  }
  if(fin_cnt < N)
    return 0;
  return 1;
}
int main(){
  //freopen("in.txt","r",stdin);
  ios_base::sync_with_stdio(false);
  int n,m;
  while(cin>>n>>m,n  ){
    for(int i=1;i<=V;i++)
      for(int j=1;j<=V;j++)
        graph[i][j]=(i==j ? 0 :INF);

    tot_cost=0;
    int t1,t2,t3;
    for(int i=0;i< n;i++){
        cin>>t1>>t2>>t3;
        graph[t1][t2]=graph[t2][t1]=t3;
      }
      init(m);

      int ret=prime(m);
      if(ret)
        cout<<tot_cost<<endl;
      else
        cout<<"?"<<endl;
  }


}

Kruskal

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define MAXN_E 100000
#define MAXN_V 100000
using namespace std;
struct Edge{
    int fm,to,dist;
}e[MAXN_E];
int fa[MAXN_V],n,m;
bool cmp(Edge a,Edge b)
{
    return a.dist < b.dist;
}
int getfa(int x){
    if(fa[x]==x) return fa[x];
    else return fa[x] = getfa(fa[x]);
}
int same(int x,int y){
    return getfa(x)==getfa(y);
}
void merge(int x,int y)
{
    int fax=getfa(x),fay=getfa(y);
    fa[fax]=fay;
}
int main()
{
    //freopen("in.txt","r",stdin);
    while(scanf("%d%d",&m,&n),m){
      for(int i=1;i<=m;i++)
      scanf("%d%d%d",&e[i].fm,&e[i].to,&e[i].dist);
      sort(e+1,e+m+1,cmp);
      for(int i=1;i<=n;i++)
          fa[i]=i;
        int rst=n,ans=0;
        for(int i=1;i<=m && rst>1;i++){
          int x=e[i].fm,y=e[i].to;
          if(same(x,y)) continue;
          else{
            merge(x,y);
            rst--;
            ans+=e[i].dist;
          }
        }
        if(rst==1)
        printf("%d\n",ans);
        else
        printf("?\n");
    }
    return 0;
}