这段时间要沉迷刷题一段时间了,就让CSDN陪我一起吧!
一、题目大意
题目的大致意思就是要你求一个图的生成树,要保证求解的这棵生成树的边的权值,最大值减去最小值最小,要求输出这个最小值。
二、题目思路以及AC代码
也很久没有刷题了(快一个月了),这道题的思路也比较直接,虽然我还并没有得到AC的结果,但我觉得思路和代码都没有问题,就先过来把博客写了,毕竟Vjudge这个评判速度也并不是很快。
对于这道题,一看生成树,肯定就要往Prim和Kruskal上去想,因为我比较熟悉Kruskal,所以就先考虑的Kruskal,对于这道题,我们可以从不同的起点跑多次Kruskal,取最终的最小值即可。
我大致解释一下这样做的原因。首先需要清楚的知道Kruskal的算法思想,就是从小边开始尝试,如果与之前的边没有成环则加入,成环则跳过,最终可以得到最小生成树。那与本题中要求得到最小极差有什么关系呢?假如我们限定这个极差中的最小值,那么使极差取得最小的生成树就是最小生成树。我们现在唯一不确定的,就是这个极差中的最小值,所以我们可以通过遍历这个极差中的最小值,跑多次Kruskal,从而获得最终结果。
这样的思路肯定没有问题,不过相对来说时间复杂度比较高,大约是O(M2),对于本题N小于100,时间给定3秒来说,应该是没有问题。
下面给出即将AC的代码:
#include <iostream>
#include <algorithm>
#define MAXN 110
#define INF 10000000
using namespace std;
struct Edge
{
/* data */
int u, v, w;
} edges[MAXN * MAXN];
int n, m;
int p[MAXN];
void init() {
for (int i = 1; i < MAXN; i++) {
p[i] = i;
}
}
int find(int x) {
if (p[x] == x) return x;
else return p[x] = find(p[x]);
}
int Kruskal(int start) {
if (m - start + 1 < n - 1) return -1;
init();
int cnt = 0;
int Max = -INF, Min = INF;
for (int i = start; i <= m; i++) {
int u = edges[i].u;
int v = edges[i].v;
int xx = find(u);
int yy = find(v);
if (xx != yy) {
Max = max(Max, edges[i].w);
Min = min(Min, edges[i].w);
p[xx] = yy;
cnt++;
}
}
if (cnt != n - 1) return -1;
return Max - Min;
}
bool cmp(Edge a, Edge b) {
return a.w < b.w;
}
int main() {
while (cin >> n >> m) {
if (!n && !m) break;
for (int i = 1; i <= m; i++) {
cin >> edges[i].u >> edges[i].v >> edges[i].w;
}
sort(edges + 1, edges + m + 1, cmp);
int res = INF;
for (int i = 1; i <= m; i++) {
int tmp = Kruskal(i);
if (tmp == -1) break;
res = min(res, tmp);
}
if (res == INF) cout << -1 << endl;
else cout << res << endl;
}
return 0;
}
如果有问题,欢迎大家指正!!!