[CSP-S模拟测试]:小Y的图(最小生成树+LCA)

题目传送门(内部题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++

猜你喜欢

转载自www.cnblogs.com/wzc521/p/11824129.html