标签:LCA,最大生成树,树上倍增
题目描述
A 国有 n 座城市,编号从 1 到 n,城市之间有 m 条双向道路。每一条道路对车辆都有重量限制,简称限重。现在有 q 辆货车在运输货物, 司机们想知道每辆车在不超过车辆限重的情况下,最多能运多重的货物。
输入输出格式
输入格式:
输入文件名为 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 。
输出格式:
输出文件名为 truck.out。
输出共有 q 行,每行一个整数,表示对于每一辆货车,它的最大载重是多少。如果货
车不能到达目的地,输出-1。
输入输出样例
输入样例#1: 复制
4 3
1 2 4
2 3 3
3 1 1
3
1 3
1 4
1 3
输出样例#1: 复制
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。
这题是在济南时晚上和zmd拼手速AC的,刺激啊,30分钟拼手速&&RP去A题
分析:先最大生成树重构树,然后对于每个询问s->t,分别查询s->LCA(s,t) 和t->LCA(s,t)路径上的最小边权。可以用树上倍增的做法记录最小边权,类似于ST表吧
话说最近模拟赛写了不少最大生成树的题,日渐变成套路。。。
Code
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#define rep(i,a,b) for(register int i=a;i<=b;i++)
#define dep(i,a,b) for(register int i=a;i>=b;i--)
#define ll long long
#define mem(x,num) memset(x,num,sizeof x)
#ifdef WIN32
#define LL "%I64d"
#else
#define LL "%lld"
#endif
using namespace std;
inline ll read()
{ll f=1,x=0;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}return x*f;
}
const int maxm=5e5+6,maxn=1e5+6,inf=0x3f3f3f;
struct edge{ll u,v,w;}e[maxm];
struct redge{ll to,next,w;}re[maxm];
int head[maxn],n,m,q,fastart[maxn],depth[maxn],fa[maxn][18],d[maxn][18],cnt=0;
bool vis[maxm];
inline bool cmp(edge x,edge y){return x.w>y.w;}
inline int find(int x){return x==fastart[x]?fastart[x]:fastart[x]=find(fastart[x]);}void dfs(int x)
{vis[x]=1;rep(i,1,16){if(depth[x]<(1<<i))break;fa[x][i]=fa[fa[x][i-1]][i-1];d[x][i]=min(d[x][i-1],d[fa[x][i-1]][i-1]);}
#define reg(x) for(int i=head[x];i;i=re[i].next)reg(x){if(vis[re[i].to])continue;fa[re[i].to][0]=x;d[re[i].to][0]=re[i].w;depth[re[i].to]=depth[x]+1;dfs(re[i].to);}
}inline int lca(int x,int y)
{if(depth[x]<depth[y])swap(x,y);int t=depth[x]-depth[y];rep(i,0,16)if((1<<i)&t)x=fa[x][i];dep(i,16,0)if(fa[x][i]!=fa[y][i]){x=fa[x][i];y=fa[y][i];}if(x==y)return x;return fa[x][0];
}int ask(int x,int father)
{int re=inf,t=depth[x]-depth[father];rep(i,0,16)if((1<<i)&t){re=min(re,d[x][i]);x=fa[x][i];}return re;
}
int main()
{mem(d,inf);n=read(),m=read();rep(i,1,m)e[i].u=read(),e[i].v=read(),e[i].w=read();sort(e+1,e+1+m,cmp);rep(i,1,n)fastart[i]=i;rep(i,1,m){int r1=find(e[i].u),r2=find(e[i].v);if(r1!=r2){fastart[r1]=r2; re[++cnt]=(redge){e[i].v,head[e[i].u],e[i].w};head[e[i].u]=cnt;re[++cnt]=(redge){e[i].u,head[e[i].v],e[i].w};head[e[i].v]=cnt;}if(cnt==2*(n-1))break; }rep(i,1,n)if(!vis[i])dfs(i);q=read();rep(i,1,q){int s=read(),t=read();int r1=find(s),r2=find(t);if(r1!=r2){printf("-1\n");continue;}else {int k=lca(s,t);printf("%lld\n",min(ask(s,k),ask(t,k)));}}return 0;
}