Codevs 3287 NOIP2013提高组 货车运输（重庆一中高2018级信息学竞赛测验2） 解题报告

【问题描述】  
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。 


【输出格式】  
　　输出共有 q 行，每行一个整数，表示对于每一辆货车，它的最大载重是多少。如果货车不能到达目的地，输出-1。


【输入输出样例】
bus.in
4 3
1 2 4
2 3 3
3 1 1
3
1 3
1 4
1 3

bus.out

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。

做题思路（正解）：根据题意，把城市看作点，道路看作无向边，限重看作边权，可以使用边集数组建立存储结构。该题的主要算法为并查集的kruskal算法，为了不超时，还应生成最大生成树，对于每次询问，先判断这两座城市是否相通，再在最大生成树上找两点之间经过路径的边的最小值，即为所求。需要注意的是，在遍历最大生成树时，要判断所选的根是否在最大生成树上。

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#include<queue>
#include<vector>
#include<cmath>
using namespace std;
const int maxn=10005;
const int inf=100000010;
int N,M,x,y,z,q;
struct edge{int u,v,w;};
vector<edge>E;
int pa[maxn],fa[maxn],d[maxn],dep[maxn];
bool used[50005],vis1[maxn];
vector<int>g[maxn],w[maxn];
void initial()  //并查集初始化
{
	for(int i=1;i<=N;i++)
	pa[i]=i;
}
bool cmp(edge aa,edge bb)  //按边权由大到小排序
{
	return aa.w>bb.w;
}
int find(int x)
{
	if(pa[x]==x)  return x;
	int root=find(pa[x]);
	pa[x]=root;
	return root;
}
bool check(int x,int y)
{
	if(find(x)==find(y))  return 1;
	return 0;
}
void bing(int x,int y)
{
	pa[find(x)]=find(y);
}
int kruskal()  //生成最大生成树
{
	initial();
	memset(used,0,sizeof(used));
	memset(vis1,0,sizeof(vis1));
	sort(E.begin(),E.end(),cmp);
	int sum=0;
	for(int i=0;i<E.size();i++)
	{
		if(!check(E[i].u,E[i].v))
		{
			bing(E[i].u,E[i].v);
			sum+=E[i].w;
			g[E[i].u].push_back(E[i].v);
			w[E[i].u].push_back(E[i].w);
			g[E[i].v].push_back(E[i].u);
			w[E[i].v].push_back(E[i].w);
			used[i]=1;
			vis1[E[i].u]=1;  //标记点在最大生成树上
			vis1[E[i].v]=1;
		}
	}
	return sum;
}
void DFS(int i,int f,int de)
{
	fa[i]=f;
	dep[i]=de;
	for(int k=0;k<g[i].size();k++)
	{
		int j=g[i][k],c=w[i][k];
		if(fa[i]==j)  continue;
		d[j]=d[i]+c;
		DFS(j,i,de+1);
	}
}
int LCA(int x,int y)  //利用LCA，找出两点的最近公共祖先，从而找出两点间经过路径的边的最小值
{
	if(dep[x]<dep[y])  swap(x,y);
	while(dep[x]!=dep[y])  x=fa[x];
	while(x!=y)  x=fa[x],y=fa[y];
	return x;
}
int getmin(int x,int y)  //<span style="font-family: Arial, Helvetica, sans-serif;">找出两点间经过路径的边的最小值</span>
{
	int z=LCA(x,y);
	int mint=inf;
	while(x!=z)
	{
		mint=min(mint,d[x]-d[fa[x]]);
		x=fa[x];
	}
	while(y!=z)
	{
		mint=min(mint,d[y]-d[fa[y]]);
		y=fa[y];
	}
	return mint;
}
int main()
{
	freopen("bus.in","r",stdin);
	freopen("bus.out","w",stdout);
	scanf("%d%d",&N,&M);
	for(int i=1;i<=M;i++)
	{
		scanf("%d%d%d",&x,&y,&z);
		E.push_back((edge){x,y,z});
	}
	int sum=kruskal();
	for(int i=1;i<=N;i++)
	if(vis1[i]==1) //确定该点在最大生成树上才进行遍历
	{
		DFS(i,i,1);
		break;
	}
	scanf("%d",&q);
	while(q--)
	{
		scanf("%d%d",&x,&y);
		if(check(x,y)==0)  printf("-1\n");  //两座城市没有连通
		else
		{
			int t=getmin(x,y);
			printf("%d\n",t);
		}
	}
	return 0;
}

考后反思：面对较大的数据规模时，要学会运用所学知识对程序进行优化，比如这道题，如果只简单用并查集来做的话，就会超时，而通过生成最大生成树，再运用LCA，找出两点间经过路径的边的最小值就可以大大节省时间。同时在小的细节（判断）上也要仔细谨慎，比如对最大生成树进行遍历时，一定要确定该点在最大生成树上才行。