畅通工程之最低成本建设问题

时间:2022-02-11 12:16:04

Think:
看了输入样例,目测是 最小生成树 问题。。而且还是模板题。。。既然是最小生成树问题,所以我就直接用了Prim算法。。初始化什么的还是老套路,直接写就可以了。。。因为最后在判断是否存在,所以也就是判断下ans是否存在就可以啦~!

某地区经过对城镇交通状况的调查,得到现有城镇间快速道路的统计数据,并提出“畅通工程”的目标:使整个地区任何两个城镇间都可以实现快速交通(但不一定有直接的快速道路相连,只要互相间接通过快速路可达即可)。现得到城镇道路统计表,表中列出了有可能建设成快速路的若干条道路的成本,求畅通工程需要的最低成本。
输入格式:

输入的第一行给出城镇数目和候选道路数目M≤3NM\le 3NM≤3N;随后的MMM行,每行给出3个正整数,分别是该条道路直接连通的两个城镇的编号(从1编号到NNN)以及该道路改建的预算成本。
输出格式:

输出畅通工程需要的最低成本。如果输入数据不足以保证畅通,则输出“Impossible”。
输入样例1:

6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3

输出样例1:

12

输入样例2:

5 4
1 2 1
2 3 2
3 1 3
4 5 4

输出样例2:

Impossible

#include<bits/stdc++.h>
#define MAX 0x3f3f3f;
using namespace std;

int Map[1050][1050];
int dis[1050];
int vis[1050];

int Prim (int n);

int main()
{
int n, m;
int i, j;
int a, b, c;
while (cin >> n >> m)
{
memset(vis, 0, sizeof(vis));
for (i = 1;i <= n;i ++)
{
for (j = 1;j <= n;j ++)
{
if (i == j)
Map[i][j] = 0;
else
Map[i][j] = MAX;
}
}
for (i = 1;i <= m;i ++)
{
cin >> a >> b >> c;
if (Map[a][b] > c)
{
Map[a][b] = Map[b][a] = c;
}
}
int ans = Prim(n);
if (ans == 4144966)
cout << "Impossible" << endl;
else
cout << ans << endl;
}
return 0;
}

int Prim (int n)
{
int i, j, t;
int sum = 0;
vis[0] = 1;
for (i = 1;i <= n;i ++)
{
for (j = 1;j <= n;j ++)
{
dis[i] = Map[i][j];
}
}
for (i = 1;i <= n;i ++)
{
t = i;
int MIN = MAX;
for (j = 1;j <= n;j ++)
{
if (MIN > dis[j] && !vis[j])
{
MIN = dis[j];
t = j;
}

}
sum = sum + MIN;
vis[t] = 1;
for (j = 1;j <= n;j ++)
{
if (Map[t][j] < dis[j]&&!vis[j])
{
dis[j] = Map[t][j];
}
}
}
return sum;
}