题目描述:
A 国有 n 座城市,编号从 1 到 n,城市之间有 m 条双向道路。每一条道路对车辆都有重量限制,简称限重。现在有 q 辆货车在运输货物,司机们想知道每辆车在不超过车辆限重的情况下,最多能运多重的货物。
输入:
第一行有两个用一个空格隔开的整数 n,m,表示 A 国有 n 座城市和 m 条道路。
接下来 m 行每行 3 个整数 x、y、z,每两个整数之间用一个空格隔开,表示从 x 号城市到 y 号城市有一条限重为 z 的道路。注意:x 不等于 y,两座城市之间可能有多条道路。
接下来一行有一个整数 q,表示有 q 辆货车需要运货。
接下来 q 行,每行两个整数 x、y,之间用一个空格隔开,表示一辆货车需要从 x 城市运输货物到 y 城市,注意:x 不等于 y。
接下来 m 行每行 3 个整数 x、y、z,每两个整数之间用一个空格隔开,表示从 x 号城市到 y 号城市有一条限重为 z 的道路。注意:x 不等于 y,两座城市之间可能有多条道路。
接下来一行有一个整数 q,表示有 q 辆货车需要运货。
接下来 q 行,每行两个整数 x、y,之间用一个空格隔开,表示一辆货车需要从 x 城市运输货物到 y 城市,注意:x 不等于 y。
输出:
输出共有 q 行,每行一个整数,表示对于每一辆货车,它的最大载重是多少。如果货车不能到达目的地,输出-1。
数据范围:
对于 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。
对于 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。
样例输入:
4 3
1 2 4
2 3 3
3 1 1
3
1 3
1 4
1 3
1 2 4
2 3 3
3 1 1
3
1 3
1 4
1 3
样例输出:
3
-1
3
分析:
从题目中给出的描述来看,很容易就能知道这是一个图论题,然而这个数据范围暴力在一秒钟之内是一定跑不出来的。
各个城市之间可能有多条道路,并且走过城市的数目增加并不会直接导致收益降低,为了保证我们能够运送的货物最多,我们可以对原图求一个最大生成树,保证城市两两之间的最小道路承重能力最大。
既然我们建立了一颗树,就可以使用求最近公共祖先的算法(LCA)来达到我们的目的了。
code:
#include<cstdio>
#include<algorithm>
#define MAXN 10000
#define MAXM 50000
#define INF 1e9
#define S 17
using namespace std;
inline int min(int a,int b){return a<b?a:b;}
struct E{
int u,v,w;
bool operator <(const E t)const{return w>t.w;}
}e[MAXM+5];
struct node{
int to,next,wei;
}adj[2*MAXM+5];
int n,m,q;
int f[MAXN+5];
int head[MAXN+5],ecnt;
int fa[MAXN+5][S+5],d[MAXN+5][S+5],depth[MAXN+5];
bool vis[MAXN+5];
inline int find(int x){return f[x]==x?x:f[x]=find(f[x]);}
void init()
{
int i;
for(i=1;i<=n;i++)
f[i]=i;
}
void add(int u,int v,int w)
{
adj[++ecnt].to=v;
adj[ecnt].wei=w;
adj[ecnt].next=head[u];
head[u]=ecnt;
adj[++ecnt].to=u;
adj[ecnt].wei=w;
adj[ecnt].next=head[v];
head[v]=ecnt;
}
void kruskal()
{
int i,tot=0;
for(i=1;i<=m&&tot<n-1;i++)
{
int u=find(e[i].u),v=find(e[i].v),w=e[i].w;
if(u==v)
continue;
add(e[i].u,e[i].v,w);
f[v]=u;
++tot;
}
}
void dfs(int u)
{
int i;
vis[u]=true;
for(i=1;i<S;i++)
{
if(depth[u]<(1<<i))
break;
fa[u][i]=fa[fa[u][i-1]][i-1];
d[u][i]=min(d[u][i-1],d[fa[u][i-1]][i-1]);
}
for(i=head[u];i;i=adj[i].next)
{
int v=adj[i].to,w=adj[i].wei;
if(!vis[v])
{
fa[v][0]=u;
d[v][0]=w;
depth[v]=depth[u]+1;
dfs(v);
}
}
}
int lca(int u,int v)
{
if(depth[u]<depth[v])
swap(u,v);
int i,t=depth[u]-depth[v];
for(i=0;i<S;i++)
if((1<<i)&t)
u=fa[u][i];
for(i=S-1;i>=0;i--)
if(fa[u][i]!=fa[v][i])
{
u=fa[u][i];
v=fa[v][i];
}
if(u==v)
return u;
return fa[u][0];
}
int query(int u,int f)
{
int i,t=depth[u]-depth[f],res=INF;
for(i=0;i<S;i++)
if((1<<i)&t)
{
res=min(res,d[u][i]);
u=fa[u][i];
}
return res;
}
int main()
{
int i,u,v;
scanf("%d%d",&n,&m);
for(i=1;i<=m;i++)
scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
init();
sort(e+1,e+m+1);
kruskal();
dfs(1);
scanf("%d",&q);
for(i=1;i<=q;i++)
{
scanf("%d%d",&u,&v);
if(find(u)!=find(v))
printf("-1\n");
else
{
int f=lca(u,v);
printf("%d\n",min(query(u,f),query(v,f)));
}
}
}