这篇博客是从旧博客 WordPress 迁移过来,内容可能存在转换异常。
这题太搞了。按照题目意思,这题只需要输入一组数据,而结果确实多组。一组就WA了。
题目传送门: http://acm.hdu.edu.cn/showproblem.php?pid=1102
分析:看到这题给出的都是矩阵形式,就知道了可以用Prim算法求MST。
最小生成树(MST)讲解请看: http://www.wutianqi.com/?p=1284 |
解答报告:
#include
using namespace std;
const int INF=1001;
int n, m;
int arcs[101][101];
int dis[101], vis[101];
int _min;
int main()
{
// freopen("input.txt", "r", stdin);
while(scanf("%d", &n;) != EOF)
{
for(int i=0; i<n; ++i)
for(int j=0; j<n; ++j)
scanf("%d", &arcs[i][j]);
scanf("%d", &m);
int a, b;
for(int i=0; i<m; ++i)
{
scanf("%d %d", &a, &b);
arcs[a-1][b-1] = arcs[b-1][a-1] = 0; // 这一步处理的好
}
memset(vis, 0, sizeof(vis));
for(int i=1; i<n; ++i)
dis[i] = arcs[0][i];
vis[0] = 1;
int sum = 0, rec = 0;
dis[0] = 0;
for(int i=1; i<n; ++i)
{
_min = INF;
for(int j=0; j<n; ++j)
if(!vis[j] && dis[j]<_min)
{
rec = j;
_min = dis[j];
}
sum += _min;
vis[rec] = 1;
for(int j=0; j<n; ++j)
if(!vis[j] && arcs[rec][j] < dis[j])
dis[j] = arcs[rec][j];
}
printf("%d\n", sum);
}
return 0;
}