PTA练习题:公路村村通

现有村落间道路的统计数据表中,列出了有可能建设成标准公路的若干条道路的成本,求使每个村落都有公路连通所需要的最低成本。

输入格式:

输入数据包括城镇数目正整数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

输出样例:

12

注:要用邻接表创建无向图,邻接矩阵容易超时

#include<stdio.h> 

int isVisited[1005]; // [0]存储路径长度,其他位置为是否访问 。 
int who[1005]; // [0]存储已选节点数,其他位置为选择的第几个节点。 

// 该结构体模拟邻接表创建图,图要创建为无向图。 
// 开始用的邻接矩阵但是要超时,无奈改为邻接表 
typedef struct Road
{
	int v[1005]; // 邻接点 
	int len[1005]; //对应邻接路径 
	int count; // 该节点邻接点数目 
}Road;
Road G[1005]; // 创建结构体数组,G[i]为第 i 个节点,结构体的数据域为与其相连的节点
			  // 例如 G[0].v[0] = 3,G[0].len[0] = 10,G[0].count = 3,
			  // 表示第 0 个节点与其相连的节点数为 G[0].count = 3个 
			  //  G[0].v[0] = 3 表示邻接 3 号节点,距离 G[0].len[0] = 10
			  // 反正用邻接表就是了,邻接矩阵要超时,当然要看具体算法 

void prim(int n) // prim算法忘却了的可以先在网上看下,最好看视频,更容易理解 
{
	who[0]++;
    who[1] = 1;
    isVisited[1] = 1;
    while(1)
    {
        int min = 100000;
        int index = -1;
        for(int i=1;i<=who[0];i++)
        {
            for(int j=1;j<=G[who[i]].count;j++)
            {
                if(G[who[i]].len[j] < min && !isVisited[G[who[i]].v[j]])
                {
                    min = G[who[i]].len[j];
                    index = G[who[i]].v[j];
                }
            }
        }
        if(index == -1)
        {
            break;
        }
        isVisited[0] += min;
        who[0]++;
        who[who[0]] = index;
        isVisited[index] = 1;
    }
    if(who[0] < n)
    {
        isVisited[0] = -1;
    }
}

int main()
{
	int n,m;
	scanf("%d %d",&n,&m);
	int x,y,l;
	for(int i=0;i<m;i++)
	{
		scanf("%d %d %d",&x,&y,&l);
		G[x].count++;
		G[y].count++;
		G[x].v[G[x].count] = y;
		G[y].v[G[y].count] = x;
		G[x].len[G[x].count] = l;
		G[y].len[G[y].count] = l;
	}
	prim(n);
	printf("%d",isVisited[0]);
	return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_45735242/article/details/106879339