题目描述:
然后想用最小树形图的方法,结果发现此题中根本行不通,只有用Kruskal算法(稍作变形)解决。
在对所有的边排序时,只需要把边的终点的高度设为第一关键字,边的长度设为第二关键字,并用Kruskal算法解决即可。
代码:
Description
a180285非常喜欢滑雪。他来到一座雪山,这里分布着M条供滑行的轨道和N个轨道之间的交点(同时也是景点),而且每个景点都有一编号i(1<=i<=N)和一高度Hi。a180285能从景点i 滑到景点j 当且仅当存在一条i 和j 之间的边,且i 的高度不小于j。 与其他滑雪爱好者不同,a180285喜欢用最短的滑行路径去访问尽量多的景点。如果仅仅访问一条路径上的景点,他会觉得数量太少。于是a180285拿出了他随身携带的时间胶囊。这是一种很神奇的药物,吃下之后可以立即回到上个经过的景点(不用移动也不被认为是a180285 滑行的距离)。请注意,这种神奇的药物是可以连续食用的,即能够回到较长时间之前到过的景点(比如上上个经过的景点和上上上个经过的景点)。 现在,a180285站在1号景点望着山下的目标,心潮澎湃。他十分想知道在不考虑时间胶囊消耗的情况下,以最短滑行距离滑到尽量多的景点的方案(即满足经过景点数最大的前提下使得滑行总距离最小)。你能帮他求出最短距离和景点数吗?
Input
输入的第一行是两个整数N,M。
接下来1行有N个整数Hi,分别表示每个景点的高度。
接下来M行,表示各个景点之间轨道分布的情况。每行3个整数,Ui,Vi,Ki。表示编号为Ui的景点和编号为Vi的景点之间有一条长度为Ki的轨道。
Output
输出一行,表示a180285最多能到达多少个景点,以及此时最短的滑行距离总和。
Sample Input
3 3
3 2 1
1 2 1
2 3 1
1 3 10
Sample Output
3 2
HINT
【数据范围】
对于30%的数据,保证 1<=N<=2000
对于100%的数据,保证 1<=N<=100000
对于所有的数据,保证 1<=M<=1000000,1<=Hi<=1000000000,1<=Ki<=1000000000。
最开始想的方法是强连通分量缩点(在同一高度用Prim求最小生成树),再对整个图乱搞,但是全错。然后想用最小树形图的方法,结果发现此题中根本行不通,只有用Kruskal算法(稍作变形)解决。
在对所有的边排序时,只需要把边的终点的高度设为第一关键字,边的长度设为第二关键字,并用Kruskal算法解决即可。
代码:
/*********************************************\
* @prob: SCOI2012 ski *
* @auth: Wang Junji *
* @stat: Accepted. *
* @date: May. 28th, 2012 *
* @memo: 图的广度优先遍历,Kruskal算法变种 *
\*********************************************/
#include <cstdio>
#include <algorithm>
typedef long long int64;
const int maxN = 100010, maxM = 1500010, SIZE = 0xfffff;
struct Edge
{
int u, v; int64 d; Edge *next; Edge() {}
Edge (int u, int v, int64 d, Edge *next):
u(u), v(v), d(d), next(next) {}
} *edge[maxN], tmp[maxM], *p; bool reach[maxN];
int q[SIZE + 1], h[maxN], F[maxN], cnt, n, m, num;
inline void Bfs()
{
int f = 0, r = 0, u, v;
for (reach[q[r++] = 1] = 1; f - r;)
for (p = edge[u = q[f++]], f &= SIZE; p; p = p -> next)
if (!reach[v = p -> v]) reach[q[r++] = v] = 1, r &= SIZE;
for (int i = 1; i < n + 1; ++i) if (reach[i]) ++num;
return;
}
int Find(int x) {return F[x] == x ? x : F[x] = Find(F[x]);}
inline bool cmp(const Edge &a, const Edge &b)
{return h[a.v] > h[b.v] || (h[a.v] == h[b.v] && a.d < b.d);}
int main()
{
freopen("ski.in", "r", stdin);
freopen("ski.out", "w", stdout);
scanf("%d%d", &n, &m);
for (int i = 1; i < n + 1; ++i) scanf("%d", h + i);
for (int i = 0; i < m; ++i)
{
int u, v, d; scanf("%d%d%d", &u, &v, &d);
if (h[u] >= h[v]) edge[u] = &(tmp[cnt++] = Edge(u, v, d, edge[u]));
if (h[v] >= h[u]) edge[v] = &(tmp[cnt++] = Edge(v, u, d, edge[v]));
}
Bfs(); int64 ans = 0;
for (int i = 1; i < n + 1; ++i) F[i] = i;
std::sort(tmp, tmp + cnt, cmp);
for (int i = 0; i < cnt; ++i)
{
if (!reach[tmp[i].u] || !reach[tmp[i].v]) continue;
int u = tmp[i].u, v = tmp[i].v, fu = Find(u), fv = Find(v);
if (fu - fv) ans += tmp[i].d, F[fu] = fv;
}
printf("%d %lld", num, ans); return 0;
}