BZOJ1832 聚會
阿新 • • 發佈:2018-04-06
pac printf can src i++ close 16px namespace 多次
Description:
Y島風景美麗宜人,氣候溫和,物產豐富。Y島上有N個城市,有N-1條城市間的道路連接著它們。每一條道路都連接某兩個城市。幸運的
是,小可可通過這些道路可以走遍Y島的所有城市。神奇的是,乘車經過每條道路所需要的費用都是一樣的。小可可,小卡卡和小YY經常
想聚會,每次聚會,他們都會選擇一個城市,使得3個人到達這個城市的總費用最小。 由於他們計劃中還會有很多次聚會,每次都選擇一
個地點是很煩人的事情,所以他們決定把這件事情交給你來完成。他們會提供給你地圖以及若幹次聚會前他們所處的位置,希望你為他們
的每一次聚會選擇一個合適的地點。
Input:
第一行兩個正整數,N和M。分別表示城市個數和聚會次數。後面有N-1行,每行用兩個正整數A和B表示編號為A和編號為B的城市之間有 一條路。城市的編號是從1到N的。再後面有M行,每行用三個正整數表示一次聚會的情況:小可可所在的城市編號,小卡卡所在的城市
編號以及小YY所在的城市編號。
Output:
一共有M行,每行兩個數Pos和Cost,用一個空格隔開。表示第i次聚會的地點選擇在編號為Pos的城市,總共的費用是經過Cost條道路所
花費的費用。
思路:思路很一眼,就是對每一組輸入,有三種情況,A走到B和C的LCA出,B走到A和C的LCA處,C走到A和B的LCA處,不過細節小心,不要寫掛
#include<bits/stdc++.h> using namespace std; const int N = 5e5 + 10View Code; int head[N], now; struct edges{ int to, next; }edge[N<<1]; void add(int u,int v){ edge[++now] = {v, head[u]}; head[u] = now;} int n, m, fa[N][22], dep[N]; void dfs(int x, int pre){ fa[x][0] = pre; for(int i = head[x]; i; i = edge[i].next){ int v = edge[i].to;if(v == pre) continue; dep[v] = dep[x] + 1; dfs(v, x); } } int LCA(int u,int v){ if(dep[v] > dep[u]) swap(u, v); int k=dep[u]-dep[v]; for(int i=0;i<=20;i++) if((1<<i)&k) u=fa[u][i]; if(u == v) return u; for(int i = 20; i >= 0; i--) if(fa[u][i] != fa[v][i]) u = fa[u][i], v = fa[v][i]; return fa[u][0]; } int main(){ scanf("%d%d",&n, &m); int x, y, z; for(int i = 1; i < n; i++){ scanf("%d%d",&x, &y); add(x, y); add(y, x); } dep[1] = 1; dfs(1, 1); for(int j = 0; j <= 20; j++) for(int i = 1; i <= n; i++) fa[i][j + 1] = fa[fa[i][j]][j]; for(int i = 1; i <= m; i++){ scanf("%d%d%d",&x, &y, &z); int lca, pos, tmp, tot = 0, ans = 1e9; lca = LCA(x, y); tot += dep[x] + dep[y] - 2*dep[lca]; tmp = LCA(lca, z); tot += dep[z] + dep[lca] - 2*dep[tmp]; if(tot < ans) ans = tot, pos = lca; swap(x, z); tot = 0; lca = LCA(x, y); tot += dep[x] + dep[y] - 2*dep[lca]; tmp = LCA(lca, z); tot += dep[z] + dep[lca] - 2*dep[tmp]; if(tot < ans) ans = tot, pos = lca; swap(y, z); tot = 0; lca = LCA(x, y); tot += dep[x] + dep[y] - 2*dep[lca]; tmp = LCA(lca, z); tot += dep[z] + dep[lca] - 2*dep[tmp]; if(tot < ans) ans = tot, pos = lca; printf("%d %d\n", pos, ans); } return 0; }
BZOJ1832 聚會