1. 程式人生 > >noip2013貨車運輸(重構樹+LCA)

noip2013貨車運輸(重構樹+LCA)

題目描述

【問題描述】
 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。

輸出

樣例輸入

4 3 
1 2 4 
2 3 3 
3 1 1 
3
1 3 
1 4 
1 3

樣例輸出

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。

題解:

對於給定的(u,v) 求u->v路徑上的最小值 使其最大。對於多個重邊,我們應該取它最大的一個,換句話說,這種邊可以刪去。

注意到m很大,必定有很多重邊。要保持其連通性,還要最大化答案,我們可以求一個最大生成樹,化簡圖。

這樣就很明朗了,我們定 1 為根節點,對於一組查詢(u,v)u->LCA(u,v)<-v路徑上的最小值。有點類似於RMQ的求解過程。

注意細節!!! 可能有多顆樹!!! 

然而我用優化讀入被卡掉了一個點。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=110000;
const int MAXM=550000;
const int M=20;
struct node{int x,y,w;}a[MAXM];
struct E{int u,v,w,next;} mm[MAXM<<1]; 
int n,m,q,cnt,h[MAXN],len,sum[MAXN][M],fa[MAXN][M],f[MAXN],dep[MAXN],vis[MAXN];

inline int rd(){
	int x=0;int f=1;char s=getchar();
	while(s>'9' || s<'0') if(s=='-')f=-1,s=getchar();
	while(s>='0' && s<='9')x=x*10+s-'0',s=getchar();
	return x*f;
}
inline void ins(int u,int v,int w){
	++len;
	mm[len].u=u;mm[len].v=v;mm[len].w=w;mm[len].next=h[u];h[u]=len;
}
bool cmp(node a,node b){
	return a.w>b.w;
}
int getfa(int x){
	if(f[x]==x) return x;
	else return f[x]=getfa(f[x]); 
} 
void dfs(int x,int f){
	fa[x][0]=f;dep[x]=dep[f]+1;vis[x]=1;
	for(int k=h[x] ; k;k=mm[k].next){
		int v=mm[k].v;int w=mm[k].w;
		if(v!=f){
			sum[v][0]=w;
			dfs(v,x);
		}
	}
}
int LCA(int x,int y){
	if(dep[x]< dep[y]) swap(x,y);
	int ans=1e18;
	for(int j=M-1;j>=0;j--){
		if(dep[fa[x][j]] >= dep[y]) ans=min(ans,sum[x][j]),x=fa[x][j];
	}
	if(x==y) return ans;
	for(int j=M-1;j>=0;j--){
		if(fa[x][j] != fa[y][j]){	
			ans=min(ans,sum[x][j]);ans=min(ans,sum[y][j]);
			x=fa[x][j];y=fa[y][j];
		}
	}
	ans=min(ans,sum[x][0]);ans=min(ans,sum[y][0]);
	return ans;
}
int main(){
//	freopen("testdata.in","r",stdin); 
//	n=rd();m=rd();
	scanf("%d%d",&n,&m);
	for(int i=1,u,v,w;i<=m;i++){
	//	a[i].x=rd(),a[i].y=rd(),a[i].w=rd();
		scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].w);
	}
	sort(a+1,a+m+1,cmp); len=cnt=0;memset(h,0,sizeof h);
	for(int i=1;i<=n;i++)f[i]=i;
	for(int i=1,aa,bb;i<=m;i++){
		aa=getfa(a[i].x);bb=getfa(a[i].y);
		if(cnt==n-1) break;
		if(aa!=bb){
			f[aa]=bb;cnt++;
			ins(a[i].x,a[i].y,a[i].w);ins(a[i].y,a[i].x,a[i].w);
		}
	}
	memset(sum,0x3f,sizeof sum);dep[0]=0;
	for(int i=1;i<=n;i++){
		if(!vis[i]) dfs(i,0);
	}
	for(int j=1;j<M;j++)
		for(int i=1;i<=n;i++){
			fa[i][j]=fa[fa[i][j-1]][j-1];
			sum[i][j]=min(sum[i][j-1] ,sum[fa[i][j-1]][j-1]);
		} 
	scanf("%d",&q);
	for(int i=1,u,v,aa,bb;i<=q;i++){
		scanf("%d%d",&u,&v);
		aa=getfa(u);bb=getfa(v);
		if(aa!=bb) printf("-1\n");
		else{
			printf("%d\n",LCA(u,v)); 
		}	
	}
	return 0;
}