7-1 公路村村通
现有村落间道路的统计数据表中,列出了有可能建设成标准公路的若干条道路的成本,求使每个村落都有公路连通所需要的最低成本。
输入格式:
输入数据包括城镇数目正整数N(≤1000)和候选道路数目M(≤3N);随后的M行对应M条道路,每行给出3个正整数,分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见,城镇从1到N编号。
输出格式:
输出村村通需要的最低成本。如果输入数据不足以保证畅通,则输出−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
#include<iostream>
using namespace std;
#define INF 0x3f3f3f3f
int n,m;
int dic[1005]={0};
int map[1005][1005]={0};
int gets(){
int res,cont,tmin,indmin;
res=cont=0;
for(int i=1;i<=n;++i){
dic[i]=map[1][i];//默认初始为1号节点 ,即默认把1号放进铺设集合
}
for(int i=2;i<=n;++i){//一共执行n-1次最小值 才能把所有的点加入集合
tmin=INF;//最小值
for(int j=2;j<=n;++j){
//find min
if(dic[j]<tmin&&dic[j]!=0){
tmin=dic[j];
indmin=j;//下标
}
}
// min
if(tmin!=INF){
dic[indmin]=0;
res+=tmin;
cont++;
//修改 dic[min][j] 集合外到集合的距离
for(int j=2;j<=n;++j){
if(map[indmin][j]<dic[j])
dic[j]=map[indmin][j];
}
}
}
return cont==n-1?res:-1;//n个节点加入集合的次数一共为n-1次
}
void init()
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
map[i][j]=INF;
}
}
}
int main(){
cin>>n>>m;
init();
int a,b,val;
for(int i=0;i<m;++i){
cin>>a>>b>>val;
map[a][b]=map[b][a]=val;
}
int res=gets();
cout<<res<<endl;
return 0;
}
#include <iostream>
#include <vector>
#include <string>
#include <climits>
using namespace std;
int n, m;
struct Map{
int Gmap[1001][1001];
}G;
int getMinPoint(int *dist, bool *visited) {
int k = -1, min = INT_MAX;
for(int i = 1; i <= n; i++) {
if(dist[i] < min && visited[i] == false) {
min = dist[i];
k = i;
}
}
return k;
}
void prim(int t) {
bool visited[n + 1];
int dist[n + 1];
int adjvec[n + 1];
for(int i = 1; i <= n; i++) {
dist[i] = G.Gmap[t][i];
visited[i] = false;
if(G.Gmap[t][i] < INT_MAX) {
adjvec[i] = t;
}else {
adjvec[i] = -1;
}
}
int k;
visited[t] = true;
dist[t] = 0;
for(int i = 1; i < n; i++) {
k = getMinPoint(dist, visited);
visited[k] = true;
for(int j = 1; j <= n; j++) {
if(dist[j] > G.Gmap[k][j] && visited[j] == false) {
dist[j] = G.Gmap[k][j];
adjvec[j] = k;
}
}
}
bool flag = false;
for(int i = 1; i <= n; i++){
if(visited[i] == true) {
flag == false;
// break;
}else {
flag = true;
break;
}
}
int count = 0;
int ans = 0;
for(int i = 1; i <= n; i++) {
ans += dist[i];
if (dist[i] != -1) {
count++;
}
}
if(count != n) {
cout << -1 << endl;
}else {
cout << ans << endl;
}
}
int main()
{
int a, b, w;
cin >> n >> m;
for(int i = 1; i <= n; i++) {
for(int j = 1; j <= n; j++) {
G.Gmap[i][j] = INT_MAX;
}
}
// while(m--) {
for(int i = 0; i < m; i++) {
cin >> a >> b >> w;
G.Gmap[a][b] = w;
G.Gmap[b][a] = w;
}
// }
prim(1);
return 0;
}

浙公网安备 33010602011771号