一道神奇的图论题……少有的Lca问题……
最开始一直有 8
个点 TLE
,后来优化了常数,发现竟然在每次询问都预处理了(emmm……这样还算“预”处理么。)。后来就是 WA
,找了半天才知道是询问的时候把 Find(x)!=Find(y)
写成了 Merge(x,y)
,愚蠢地合并了 x
和 y
……
不过值得欣慰的是,最后耗时只有 288ms
,但还是写了一下午,故发篇题解冷静一下。
Description
A 国有 n 座城市,编号从 1 到 n,城市之间有 m 条双向道路。每一条道路对车辆都有重量限制,简称限重。现在有 q 辆货车在运输货物, 司机们想知道每辆车在不超过车辆限重的情况下,最多能运多重的货物。
Input
输入文件名为 truck.in。
输入文件第一行有两个用一个空格隔开的整数 n,m,表示 A 国有 n 座城市和 m 条道
路。 接下来 m 行每行 3 个整数 x、 y、 z,每两个整数之间用一个空格隔开,表示从 x 号城市到 y 号城市有一条限重为 z 的道路。注意: x 不等于 y,两座城市之间可能有多条道路 。
接下来一行有一个整数 q,表示有 q 辆货车需要运货。
接下来 q 行,每行两个整数 x、y,之间用一个空格隔开,表示一辆货车需要从 x 城市运输货物到 y 城市,注意: x 不等于 y 。
Output
输出文件名为 truck.out。
输出共有 q 行,每行一个整数,表示对于每一辆货车,它的最大载重是多少。如果货
车不能到达目的地,输出-1。
Sample Input
4 3
1 2 4
2 3 3
3 1 1
3
1 3
1 4
1 3
Sample Output
3
-1
3
数据范围及提示
对于 30%的数据,0 < n < 1,000,0 < m < 10,000,0 < q< 1,000;
对于 60%的数据,0 < n < 1,000,0 < m < 50,000,0 < q< 1,000;
对于 100%的数据,0 < n < 10,000,0 < m < 50,000,0 < q< 30,000,0 ≤ z ≤ 100,000。
正解:生成树+lca
看题!题目要求我们求出 x->y
路径中最大边权的最小值,所以可以建一棵“最大生成树”。易证最大生成树上的路径边权最小值为所有路径中最大。
求出了最大生成树,就可以在树上求两点间路径的最小值了。这时就可以求出两点的lca,则 ans=min(dis(x,lca), dis(y,lca))
。
具体地,最大生成树怎么求呢?用kruskal比较划算。在边排序时从大到小排就好了。至于lca,个人比较喜欢用倍增。
代码:
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
const int Inf = 2139062143;
int N, M, Q, X, Y;
int G[10010], F[10010][35];
int st[10010], tot;
int dep[10010], dis[10010];
struct HH{int u,v,w;}H[50010];
struct EE{int to,val,nxt;}E[50010];
int read(){
int x=0, f=1; char ch = getchar();
while(ch<'0' || ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0' && ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
return x*f;
}
void Add(int x, int y, int z){
E[++tot].to=y;
E[tot].val=z;
E[tot].nxt=st[x];
st[x]=tot;
}
bool cmp(HH a, HH b){return a.w > b.w;}
int Find(int x){return G[x]==x ? x : G[x]=Find(G[x]);}
bool Merge(int x, int y){
int xx=Find(x), yy=Find(y);
if(xx != yy){G[yy]=xx;return 1;}
return 0;
}
void Kruskal(){
sort(H+1, H+M+1, cmp);
for(int i = 1; i <= N; i++) G[i]=i;
int i=1, cnt=0;
while(i<=M && cnt!=N-1){
if(Merge(H[i].u, H[i].v)){
cnt++;
Add(H[i].u,H[i].v,H[i].w);
Add(H[i].v,H[i].u,H[i].w);
}
i++;
}
}
void Dfs(int x){
for(int i = st[x]; i; i = E[i].nxt){
if(E[i].to == F[x][0]) continue;
dep[E[i].to] = dep[x]+1;
F[E[i].to][0] = x;
dis[E[i].to] = E[i].val;
Dfs(E[i].to);
}
}
void Init(){
for(int i = 1; i <= 15; i++){
for(int j = 1; j <= N; j++){
F[j][i] = F[F[j][i-1]][i-1];
}
}
}
int Lca(int x, int y){
if(dep[x] < dep[y]) swap(x, y);
for(int i = 15; i >= 0; i--){
if(dep[F[x][i]] >= dep[y]) x = F[x][i];
}
if(x == y) return x;
for(int i = 15; i >= 0; i--){
if(F[x][i] != F[y][i]){
x = F[x][i]; y = F[y][i];
}
}
return F[x][0];
}
int Query(int x, int y){
int t = Lca(x, y);
int a=Inf, b=Inf;
for(int i = x; i != t; i = F[i][0]) if(dis[i]<a) a=dis[i];
for(int i = y; i != t; i = F[i][0]) if(dis[i]<b) b=dis[i];
return a<b ? a : b;
}
int main()
{
N=read(); M=read();
for(int i = 1; i <= M; i++){
H[i].u=read();H[i].v=read();H[i].w=read();
}
Kruskal(); Q=read();
memset(dis, 127, sizeof(dis));
dep[1]=1; Dfs(1); Init();
while(Q--){
X=read(); Y=read();int ans;
if(Find(X)!=Find(Y)) ans = Inf;
else ans = Query(X, Y);
printf("%d\n", ans==Inf ? -1 : ans);
}
return 0;
}