题目描述
AAA国有nn n座城市,编号从 11 1到n nn,城市之间有 mmm 条双向道路。每一条道路对车辆都有重量限制,简称限重。现在有 qqq 辆货车在运输货物, 司机们想知道每辆车在不超过车辆限重的情况下,最多能运多重的货物。
输入输出格式
输入格式:第一行有两个用一个空格隔开的整数n,m n,mn,m,表示 AAA 国有n nn 座城市和 mmm 条道路。
接下来 mmm行每行3 3 3个整数 x,y,zx, y, zx,y,z,每两个整数之间用一个空格隔开,表示从 xx x号城市到y y y号城市有一条限重为 zzz 的道路。注意: xxx 不等于 yyy,两座城市之间可能有多条道路 。
接下来一行有一个整数 q,表示有 q 辆货车需要运货。
接下来 q 行,每行两个整数 x、y,之间用一个空格隔开,表示一辆货车需要从 x 城市运输货物到 y 城市,注意: x 不等于 y 。
输出格式:共有 qqq 行,每行一个整数,表示对于每一辆货车,它的最大载重是多少。如果货车不能到达目的地,输出−1-1−1。
输入输出样例
说明
对于 30%30\%30%的数据,0<n<1,000,0<m<10,000,0<q<1,0000 < n < 1,000,0 < m < 10,000,0 < q< 1,0000<n<1,000,0<m<10,000,0<q<1,000;
对于 60%60\%60%的数据,0<n<1,000,0<m<50,000,0<q<1,0000 < n < 1,000,0 < m < 50,000,0 < q< 1,0000<n<1,000,0<m<50,000,0<q<1,000;
对于 100%100\%100%的数据,0<n<10,000,0<m<50,000,0<q<30,000,0≤z≤100,0000 < n < 10,000,0 < m < 50,000,0 < q< 30,000,0 ≤ z ≤ 100,0000<n<10,000,0<m<50,000,0<q<30,000,0≤z≤100,000。
这题暴力算法可以想到O(n2),但是我们可以往贪心的方向想。
比如有些小边实际上根本就不需要走,于是我们想到最大生成树。
求了最大生成树之后呢,我们可以用lca来查找两个节点的路径上最小的边,于是这道题就解决了。
ps:我做这道题的时候一开始做的麻烦了,多做了一次dfs并且想把每颗最大生成树分开来,其实完全没有必要。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define ll long long
#define il inline
#define db double
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
using namespace std;
il int gi()
{
int x=0,y=1;
char ch=getchar();
while(ch<'0'||ch>'9')
{
if(ch=='-')
y=-1;
ch=getchar();
}
while(ch>='0'&&ch<='9')
{
x=x*10+ch-'0';
ch=getchar();
}
return x*y;
}
struct edge
{
int next,to,from,w;
}e[100045],ne[100045],ee[100045];
int head[100045],cnt,fa[100045],treenum,f[10045][21],minx[10045][21],deep[10045],headd[100045],cnt1;
bool vis[100045];
il int find(int x)
{
if(fa[x]!=x)
fa[x]=find(fa[x]);
return fa[x];
}
il void add(int from,int to,int w)
{
ne[++cnt1].next=headd[from];
ne[cnt1].to=to;
ne[cnt1].w=w;
headd[from]=cnt1;
}
il bool cmp(edge a,edge b)
{
return a.w>b.w;
}
il int lca(int x,int y)
{
int ans=1e9;
if(deep[y]>deep[x])
swap(x,y);
for(int i=20;i>=0;i--)
if(deep[f[x][i]]>=deep[y])
ans=min(ans,minx[x][i]),x=f[x][i];
if(x!=y)
{
for(int i=20;i>=0;i--)
if(f[x][i]!=f[y][i])
ans=min(min(ans,minx[x][i]),minx[y][i]),x=f[x][i],y=f[y][i];
ans=min(ans,min(minx[x][0],minx[y][0]));
}
return ans;
}
il void dfs(int x)
{
int r=headd[x],now;
while(r!=-1)
{
now=ne[r].to;
if(!vis[now])
{
vis[now]=1;
deep[now]=deep[x]+1;
f[now][0]=x;
minx[now][0]=ne[r].w;
dfs(now);
}
r=ne[r].next;
}
}
int main()
{
memset(head,-1,sizeof(head));
memset(headd,-1,sizeof(headd));
memset(minx,127/3,sizeof(minx));
int n=gi(),m=gi();
int x,y,z;
for(int i=1;i<=m;i++)
{
x=gi(),y=gi(),z=gi();
ee[i].from=x;
ee[i].to=y;
ee[i].w=z;
}
for(int i=1;i<=n;i++)
fa[i]=i;
sort(ee+1,ee+1+m,cmp);
for(int i=1;i<=m;i++)
{
x=ee[i].from,y=ee[i].to,z=ee[i].w;
int r1=find(x),r2=find(y);
if(r1!=r2)
{
fa[r2]=r1;
add(x,y,z);
add(y,x,z);
}
}
for(int i=1;i<=n;i++)
if(fa[i]==i)
vis[i]=1,deep[i]=1,dfs(i);
for(int j=1;j<=20;j++)
for(int i=1;i<=n;i++)
{
f[i][j]=f[f[i][j-1]][j-1];
minx[i][j]=min(minx[i][j-1],minx[f[i][j-1]][j-1]);
}
int q=gi();
for(int i=1;i<=q;i++)
{
x=gi(),y=gi();
if(find(x)==find(y))
printf("%d\n",lca(x,y));
else
printf("-1\n");
}
return 0;
}