今天写最下生成树prime算法实现及1233的解题思路
最小生成树prim算法。
prim算法的原理和最短路的原理非常相似,① 要建立一个二维数组保存数据;② 找到一个点(这个随意,因为要经过所有的点,所以从哪个点开始并不重要),并把最小的权值(距离)保存进数组dis[ ]里;③ 从二维数组中找到与上一个点相连的权值最小的点, 并与上一点的权值相比较,权值比上一个点的权值小就保存入dis[ ]中;④ 用循环执行③步骤n-1(n为最小生成树中点的个数)次。
hdu 1233 的解题思路。
先附上原题:
还是畅通工程
Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)Total Submission(s): 24052 Accepted Submission(s): 10684
Problem Description
某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 );随后的N(N-1)/2行对应村庄间的距离,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间的距离。为简单起见,村庄从1到N编号。
当N为0时,输入结束,该用例不被处理。
当N为0时,输入结束,该用例不被处理。
Output
对每个测试用例,在1行里输出最小的公路总长度。
Sample Input
3 1 2 1 1 3 2 2 3 4 4 1 2 1 1 3 4 1 4 1 2 3 3 2 4 2 3 4 5 0
Sample Output
3 5
首先是数据的输入及数组的处理,要保存数据就必须先创建二维数组保存数据(要注意的是从a地到a地的权值为0),这个没难度,直接附代码!
const int maxx = 102;
int map[maxx][maxx];
int main()
{
int n;
int m;
while(scanf("%d",&n)&&n)
{
m = ((n-1)*n)/2;
int a, b, c;
for(int i=1; i<=n; i++)
{
for(int j=1; j<=n; j++)
{
if(i==j) map[i][j] = 0;
else map[i][j] = inf;
}
}
for(int i=1; i<=m; i++)
{
scanf("%d%d%d",&a,&b,&c);
if(map[a][b]>c)
{
map[a][b] = map[b][a] = c;
}
}
int d = prim(n);
printf("%d\n",d);
}
return 0;
}
然后代码中的prim即核心内容,接下来是最想生成树prim算法在hdu 1233的实现。
因为题目要求是输出最的小公路长度,所以我们在prim中只要返回总长度就行了,于是我们就定义sum来记录公路的总长度,因为要标记点是否被访问过,所以,我们定义bool vis[ ]来记录点是否被访问过,当然我们也要定义一个minn来保存最小的权值并把它加入到sum中!
最后一些思路实现就写在代码的注释里了!
代码:
const int inf = 0xfffff;
const int maxx = 102;
int dis[maxx];
bool vis[maxx];
int prim(int n)
{
int sum=0; ///sum记得要初始化为0;
memset(vis,0,sizeof(vis)); ///标记是否被访问过要记得初始化为0;
for(int i=1; i<=n; i++)
{
dis[i] = map[1][i]; ///把与点①相连的点的权值放入dis[ ]中 ;
}
vis[1] = 1; ///标记点①已经被访问过;
for(int i=1; i<n; i++)
{
int minn = inf, k; ///定义最小权值为最大值,方便更新minn;
for(int j=1; j<=n; j++)
{
if(!vis[j]&&minn>dis[j]) ///判断点是否被访问过;
{
minn = dis[j]; ///记录最小权值;
k = j; ///记录最小权值的点;
}
}
sum = sum + minn; /// 把最小的权值加入sum中;
vis[k] = 1; ///标记最小权值的点为已访问过;
for(int j=1; j<=n; j++)
{
if(dis[j]>map[k][j]) ///判断大小,并决定是否要更新最小生成树;
dis[j] = map[k][j];
}
}
return sum;
}
其中最重要的两个语句是:
if(!vis[j]&&minn>dis[j])
和
if(dis[j]>map[k][j]) dis[j] = map[k][j];
以上!