histcat

histcat

P1967 [NOIP2013 提高組] 貨車運輸

題目#

貨車運輸#

A 國有 $n$ 座城市,編號從 $1$ 到 $n$,城市之間有 $m$ 條雙向道路。每一條道路對車輛都有重量限制,簡稱限重。

現在有 $q$ 輛貨車在運輸貨物, 司機們想知道每輛車在不超過車輛限重的情況下,最多能運多重的貨物。

輸入格式#

第一行有兩個用一個空格隔開的整數 $ n,m$,表示 A 國有 $ n$ 座城市和 $m$ 條道路。

接下來 $m$ 行每行三個整數 $x, y, z$,每兩個整數之間用一個空格隔開,表示從 $x $ 號城市到 $ y $ 號城市有一條限重為 $z$ 的道路。
注意: $x \neq y$,兩座城市之間可能有多條道路 。

接下來一行有一個整數 $q$,表示有 $q$ 輛貨車需要運貨。

接下來 $q$ 行,每行兩個整數 $x,y$,之間用一個空格隔開,表示一輛貨車需要從 $x$ 城市運輸貨物到 $y$ 城市,保證 $x \neq y$

對於 $100%$ 的數據,$1 \le n < 10^4$,$1 \le m < 5\times 10^4$,$1 \le q< 3\times 10^4 $,$0 \le z \le 10^5$。

題解#

由於要求路徑上的最小值最大,所以~二分(當然可以,但我不會 QAQ)~ 求出最大生成樹來,然後對於每次詢問的兩點 lca 查詢路徑上的最小值即可

具體實現和 lca 差不多

問題#

1. 建雙向邊

2. 並查集fa[/*!!!*/x_fa] = y;

3.125 行取 min 寫成了 anc

代碼#

	#include<bits/stdc++.h>
	using namespace std;

	int read()
	{
		int 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 f * x;
	}

	const int N = 1e4 + 10, M = 1e5 + 100;

	struct U
	{
		int x, y, z;
		bool operator < (const U o) const
		{
			return z > o.z;
		}
	}u[M >> 1]; int cnt = 1;

	int fa[N], n, m;

	int anc[N][20], path_min[N][20], depth[N];

	int getfa(int u)
	{
		if(fa[u] == u) return u;
		return fa[u] = getfa(fa[u]);
	}

	void merge(int x, int y)
	{
		int x_fa = getfa(x), y_fa = getfa(y);
		fa[/*!!!*/x_fa] = y;
	}

	int head[N], nxt[M], to[M], edge[M], tot = 1;

	void add(int x, int y, int z)
	{
		to[++tot] = y;
		edge[tot] = z;
		nxt[tot] = head[x];
		head[x] = tot;
	}

	void dfs(int u, int fa)
	{
		depth[u] = depth[fa] + 1;
		anc[u][0] = fa;
		for(int i = head[u]; i; i = nxt[i])
		{
			int v = to[i];
			if(v == fa) continue;
			path_min[v][0] = edge[i];
			dfs(v, u);
		}
	}

	void lca_init()
	{
		for(int j = 1;j < 20;j++)
		{
			for(int i = 1;i <= n;i++)
			{
				anc[i][j] = anc[anc[i][j - 1]][j - 1];
				path_min[i][j] = min(path_min[i][j - 1], path_min[anc[i][j - 1]][j - 1]);
			}
		}
	}

	int lca_query_anc(int x, int y)
	{
		if(depth[x] < depth[y])
			swap(x, y);
		for(int i = 19;i >= 0;i--)
		{
			if(depth[anc[x][i]] >= depth[y])
				x = anc[x][i];
		}
	
		if(x == y) return x;
	
		for(int i = 19;i >= 0;i--)
		{
			if(anc[x][i] != anc[y][i])
				x = anc[x][i], y = anc[y][i];
		}
		return anc[x][0];
	}

	int lca_query_min(int x, int y)
	{
		int ans = 0x3f3f3f3f;
		if(depth[x] < depth[y])
			swap(x, y);
		for(int i = 19;i >= 0;i--)
		{
			if(depth[anc[x][i]] >= depth[y])
			{
				ans = min(ans, path_min[x][i]);	x = anc[x][i]; 
			}
		}
		if(x == y) return ans;
		for(int i = 19;i >= 0;i--)
		{
			if(anc[x][i] != anc[y][i])
			{
				ans = min(ans, path_min[x][i]), ans = min(ans, path_min[y][i]);
				x = anc[x][i], y = anc[y][i];
			}
		}
		return min(ans, min(path_min[x][0], path_min[y][0]));
	}

	int main()
	{
		n = read(), m = read();
		for(int i = 1;i <= n;i++)
		{
			fa[i] = i;
		}
	
		int x, y, z;
		for(int i = 1;i <= m;i++)
		{
			x = read(), y = read(), z = read();
			u[cnt].x = x;
			u[cnt].y = y, u[cnt].z = z;
			++cnt;
		}
	
		sort(u + 1, u + 1 + m);
	
		for(int i = 1;i <= m;i++)
		{
			int x_fa = getfa(u[i].x), y_fa = getfa(u[i].y);
			if(x_fa != y_fa)
			{
				merge(u[i].x, u[i].y);
				add(u[i].x, u[i].y, u[i].z);
				add(u[i].y, u[i].x, u[i].z);
			}
		}
	
		for(int i = 1;i <= n;i++)
		{
			if(fa[i] == i)
			{
				dfs(i, 0);
			}
		}
		lca_init();
		int q;
		q = read();
	
//		for(int i = 1;i <= n;i++)
//		{
//			cout << "qwq "<<path_min[i][2] << endl;
//		}
	
		while(q--)
		{
			x = read(), y = read();
			int x_fa = getfa(x), y_fa = getfa(y);
			if(x_fa != y_fa)
			{
				printf("-1\n");
				continue;
			}
			cout << lca_query_min(x, y) << endl; 
	}

}
載入中......
此文章數據所有權由區塊鏈加密技術和智能合約保障僅歸創作者所有。