题目传送门(内部题131)
输入格式
第一行三个整数$n$、$m$和$Q$。
接下来$m$行每行三个整数$x$、$y$、$z$($1\leqslant x,y\leqslant n,1\leqslant z\leqslant 1,000,000$),表示有一条连接$x$和$y$长度为$z$的边。
接下来$Q$行每行两个整数$x$、$y$($x\neq y$),表示一组询问。
输出格式
$Q$行每行一个整数,表示一组询问的答案。
样例
样例输入:
5 5 4
1 2 3
1 3 2
3 2 1
1 4 5
2 4 4
1 2
1 4
3 5
2 4
样例输出:
2
4
-1
4
数据范围与提示
对于前$30\%$的测试数据,满足$1\leqslant n,m,Q\leqslant 1,000$。
对于另外$30\%$的测试数据,保证图联通。
对于$100\%$的测试数据,满足$1\leqslant n,m,Q\leqslant 300,000$。
对于$100\%$的测试数据,保证不存在自环,但可能存在重边。
请使用$scanf,printf$或速度更快的读入输出方式。
题解
有人问我$30\%$的暴力怎么打(问题是$ta$还$A$了)……
那我就简单说一下。
最短路思想,用$Dijkstra$,将原本的$dis[v]=dis[u]+e[i].w$改成$dis[v]=\max(dis[u],e[i].w)$就好了。
千万不要想当然,比方说下面这份代码$\downarrow$
认真看一下,虽说时间复杂度是对的,但是如果如下面这张图$\downarrow$
我们可能会选择$x\stackrel{2}{\rightarrow}o\stackrel{1}{\rightarrow}y$这条路径;然而当发现$x\stackrel{1}{\rightarrow}o$更优时会发现$o\rightarrow y$已经走过了,就不会再更新答案,这也就是为什么最短路不是这么求。
现在来说正解吧,先来考虑联通的情况。
这个最优路径上的所有边一定位于最小生成树上,所以可以求$x,y$到$lca$上的最长边即可。
不联通的情况也无非就是记录一下两个点在不在一个联通块内即可。
时间复杂度:$\Theta(m\log m+q\log n)$。
期望得分:$100$分。
实际得分:$100$分。
代码时刻
#include<bits/stdc++.h> using namespace std; struct node{int x,y,z;bool d;}b[300001]; struct rec{int nxt,to,w;}e[600001]; int head[300001],cnt,tot; int n,m,Q; int f[300001],depth[300001],bel[300001],fa[300001][21],mi[300001][21]; bool cmp(node a,node b){return a.z<b.z;} int find(int x){return x==f[x]?x:f[x]=find(f[x]);} void add(int x,int y,int w) { e[++cnt].nxt=head[x]; e[cnt].to=y; e[cnt].w=w; head[x]=cnt; } void dfs(int x) { bel[x]=tot; for(int i=head[x];i;i=e[i].nxt) { if(depth[e[i].to])continue; depth[e[i].to]=depth[x]+1; fa[e[i].to][0]=x; mi[e[i].to][0]=e[i].w; for(int j=1;j<=20;j++) { fa[e[i].to][j]=fa[fa[e[i].to][j-1]][j-1]; mi[e[i].to][j]=max(mi[e[i].to][j-1],mi[fa[e[i].to][j-1]][j-1]); } dfs(e[i].to); } } int LCA(int x,int y) { if(depth[x]>depth[y])swap(x,y); int res=0; for(int i=20;i>=0;i--) if(depth[fa[y][i]]>=depth[x]) { res=max(res,mi[y][i]); y=fa[y][i]; } if(x==y)return res; for(int i=20;i>=0;i--) if(fa[x][i]!=fa[y][i]) { res=max(res,max(mi[x][i],mi[y][i])); x=fa[x][i];y=fa[y][i]; } return max(res,max(mi[x][0],mi[y][0])); } int main() { scanf("%d%d%d",&n,&m,&Q); for(int i=1;i<=n;i++)f[i]=i; for(int i=1;i<=m;i++)scanf("%d%d%d",&b[i].x,&b[i].y,&b[i].z); sort(b+1,b+m+1,cmp); for(int i=1;i<=m;i++) { int x=find(b[i].x); int y=find(b[i].y); if(x==y)continue; b[i].d=1; f[y]=x; } for(int i=1;i<=m;i++) if(b[i].d) { add(b[i].x,b[i].y,b[i].z); add(b[i].y,b[i].x,b[i].z); } for(int i=1;i<=n;i++) if(!depth[i]) { tot++; depth[i]=1; dfs(i); } while(Q--) { int x,y; scanf("%d%d",&x,&y); if(bel[x]!=bel[y])puts("-1"); else printf("%d\n",LCA(x,y)); } return 0; }
rp++
原文地址:https://www.cnblogs.com/wzc521/p/11824129.html