图算法—Problem A
题意
题目大意就是输入一个N表示有n个村庄,然后输入n行n列,第i行j个元素表示i村庄离j村庄的路数多远(所以i行i列一般为零),然后输入q,表示q条路已修,然后输入q对数就可以了。(权值为零)。求使其n个村庄最小的连通值。
解题思路
最小生成树问题。看PPT上有Prim算法和Kruskal算法两种。这道题采用的是Prim算法。写的过程中,需要注意的一点就是可能会有重复的路出现,但是有不同的值。既然是求最小生成树,那么可以用prim算法在输入路的时候保证是最小的值。
感想
不太会啊,抓狂!
AC代码
#include <cstdio>
#include <iostream>
#include <algorithm>
#include<cstring>
using namespace std;
const int N = 101;
int map[N][N];
int mark[N];
int n,q,i,j;
int Prim()
{
int sum = 0;
int t = n;
int min,k;
memset(mark,0,sizeof(mark));
while(--t)
{
min = 10000;
for (i = 2; i <= n; i++)
{
if(mark[i]!=1 && map[1][i] < min)
{
min = map[1][i];
k = i;
}
}
mark[k] = 1;
sum += min;
for (j = 2; j <= n; j++)
{
if(mark[j]!=1 && map[k][j] < map[1][j])
{
map[1][j] = map[k][j];
}
}
}
return sum;
}
int main()
{
while(cin>>n)
{
for (i = 1; i <= n; i++)
{
for (j = 1; j <= n; j++)
{
cin>>map[i][j];
}
}
cin>>q;
while(q--)
{
cin>>i>>j;
map[i][j] = 0;
map[j][i] = 0;
}
cout<<Prim()<<endl;
}
return 0;
}