畅通工程
Time Limit: 1000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)Total Submission(s): 34565 Accepted Submission(s): 15320
Problem Description
省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。经过调查评估,得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序,计算出全省畅通需要的最低成本。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M ( < 100 );随后的 N
行对应村庄间道路的成本,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间道路的成本(也是正整数)。为简单起见,村庄从1到M编号。当N为0时,全部输入结束,相应的结果不要输出。
行对应村庄间道路的成本,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间道路的成本(也是正整数)。为简单起见,村庄从1到M编号。当N为0时,全部输入结束,相应的结果不要输出。
Output
对每个测试用例,在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通,则输出“?”。
Sample Input
3 31 2 11 3 22 3 41 32 3 20 100
Sample Output
3?
Source
解题思路:我们先按照修路的费用从低到高给所有的路排一个序,然后按照费用从低到高修建路,最后修建的路能连通所有城市就输出费用,不能连通就输出“?”即可。
AC代码(Prim):
没有优化的PrimO(n^2)
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<string>
#include<math.h>
#include<stdlib.h>
#include<queue>
#include<map>
#include<set>
#define bug printf("*********\n");
#define mem0(a) memset(a, 0, sizeof(a));
#define mem1(a) memset(a, -1, sizeof(a));
#define in1(a) scanf("%d" ,&a);
#define in2(a, b) scanf("%d%d", &a, &b);
#define out1(a) printf("%d\n", a);
#define out2(a, b) printf("%d %d\n", a, b);
#define pb(G, a, b) G[a].push_back(b);
using namespace std;
typedef long long LL;
typedef pair<int, int> par;
const int mod = 1e9+7;
const int INF = 1e9+7;
const int N = 1000010;
const double pi = 3.1415926;
int n, m, ans;
int way[110][110], vis[110], cost[110]; //cost代表连接到生成树中的最小花费
void prim()
{
int cur = 1;
int sum = 0;
cost[cur] = 0;
while(sum < m) {
cur = -1;
for(int i = 1; i <= m; i ++) { //寻求最小花费的先边
if(!vis[i] && (cur == -1 || cost[i] < cost[cur])) {
cur = i;
}
}
vis[cur] = 1;
ans += cost[cur];
sum ++;
if(ans >= INF) break; //没有可以连的边了
for(int i = 1; i <= m; i ++) { //更新边权最小值
cost[i] = min(cost[i], way[cur][i]);
}
}
}
int main()
{
int x, y, z;
while(~scanf("%d%d", &n, &m) && n) {
mem0(vis);
ans =0;
fill(way[0], way[110], INF);
fill(cost, cost+110, INF);
for(int i = 0; i < n; i ++) {
scanf("%d%d%d", &x, &y, &z);
way[x][y] = way[y][x] = z;
}
prim();
if(ans < INF) printf("%d\n", ans);
else printf("?\n");
}
return 0;
}
优先队列优化的PrimO(n*logn)
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
#include<string>
#include<math.h>
#include<stdlib.h>
#include<queue>
#include<map>
#include<set>
#define bug printf("*********\n");
#define mem0(a) memset(a, 0, sizeof(a));
#define mem1(a) memset(a, -1, sizeof(a));
#define in1(a) scanf("%d" ,&a);
#define in2(a, b) scanf("%d%d", &a, &b);
#define out1(a) printf("%d\n", a);
#define out2(a, b) printf("%d %d\n", a, b);
#define pb(G, a, b) G[a].push_back(b);
using namespace std;
typedef long long LL;
typedef pair<int, int> par;
const int mod = 1e9+7;
const int INF = 1e9+7;
const int N = 1000010;
const double pi = 3.1415926;
int n, m, ans;
int way[110][110], vis[110], cost[110]; //cost代表连接到生成树中的最小花费
bool prim()
{
int sum = 0;
cost[1] = 0;
priority_queue<pair<int, int> > q;
q.push(make_pair(0, 1));
while(!q.empty()) {
par cur = q.top(); //logn时间内得出的最小值
q.pop();
if(vis[cur.second]) continue;
vis[cur.second] = 1;
ans += -cur.first;
sum ++;
for(int i = 1; i <= m; i ++) {
cost[i] = min(cost[i], way[cur.second][i]);
if(!vis[i] && cost[i] < INF) q.push(make_pair(-cost[i], i));
}
}
return sum == m;
}
int main()
{
int x, y, z;
while(~scanf("%d%d", &n, &m) && n) {
mem0(vis);
ans =0;
fill(way[0], way[110], INF);
fill(cost, cost+110, INF);
for(int i = 0; i < n; i ++) {
scanf("%d%d%d", &x, &y, &z);
way[x][y] = way[y][x] = z;
}
if(prim()) printf("%d\n", ans);
else printf("?\n");
}
return 0;
}
AC代码(Kruskal):O(n*logn)
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<math.h>
#include<time.h>
#include<map>
#include<string>
#include<algorithm>
#include<set>
#define N 110
using namespace std;
int root[N], sum, money = 0;
struct node
{
int a;
int b;
int c;
}road[N]; //连接a-b城市的路修建费用为c
bool cmp(node a, node b) //按照费用从低到高排序
{
return a.c < b.c;
}
int formalset(int n) //初始化
{
for(int i = 1; i <= n ;i ++)
{
root[i] = i;
}
}
int find_root(int k) //查找根
{
if(root[k] != k)
root[k] = find_root(root[k]);
return root[k];
}
void merge_root(int a, int b, int c) //两城市连通,合并,并加上费用
{
if(find_root(a) == find_root(b)) //已经连通,返回
return;
sum --;
money += c;
root[find_root(b)] = find_root(a); //没有连通,合并连通
}
int main()
{
int m, n;
while(~scanf("%d", &m))
{
if(!m) break;
money = 0;
scanf("%d", &n);
sum = n - 1;
formalset(n);
for(int i = 0; i < m; i ++)
{
scanf("%d%d%d", &road[i].a, &road[i].b, &road[i].c);
}
sort(road, road+m, cmp);
for(int i = 0; i < m; i ++)
{
if(sum == 0) break; //如果所有的城市连通,就退出
merge_root(road[i].a, road[i].b, road[i].c);
}
if(sum) printf("?\n");
else printf("%d\n",money);
}
return 0;
}