Problem Description
给你 n 个点,接下来 n-1 行,每行输入 u, v 代表 u 点和 v 点之间有一条无向边。
接下来输入 q,代表有 q 次询问。
u 到 v 的道路中,距离 1 点最近的点,是重要的点。
每次询问输入 u, v 问你。u 到 v 重要的点,u+1 到 v 重要的点…. v-1 到 v 重要的点,那个点距离 1 更近,输出这个点。
思路:
就是以 1 为根,求 lca(u, v), lca(u+1,v )… lca(v-1, v) 距离 1 最近的点。
lca(u,v) 代表u->v路径中,深度最低的点,也就是距离1最近的点。
2->3的路径 和 3->4的路径 和 4->5的路径 包含了 2->5的路径
假设我们求出 lca(2, 3), lca(3, 4)。那么我们就可以知道 2->4, 3->4那个点距离1更近, 也就可以知道询问u,v的结果。
所以预处理求出lca(1, 2), lca(2, 3), …, lca(n-1, n);
然后用线段树维护u,v结果。
#include<cstdio>
#include<vector>
using namespace std;
const int N = 2*1e5+100;
const int LOG_N = 30;
#define MID int mid = (l+r)>>1
#define lson root<<1
#define rson root<<1|1
struct node
{
int depth, id;
};
node tree[4*N];
vector<int> G[N];
int parent[LOG_N][N];
int depth[N], lc[N];
void dfs(int v, int p, int d)
{
parent[0][v] = p;
depth[v] = d;
for(int i = 0; i < G[v].size(); i++) {
if(G[v][i] != p) dfs(G[v][i], v, d+1);
}
}
void init(int V)
{
dfs(1, -1, 0);
for(int k = 0; k+1 < LOG_N; k++) {
for(int v = 1; v <= V; v++) {
if(parent[k][v] < 0) parent[k+1][v] = -1;
else parent[k+1][v] = parent[k][parent[k][v]];
}
}
}
int lca(int u, int v)
{
if(depth[u] > depth[v]) swap(u, v);
for(int k = 0; k < LOG_N; k++) {
if((depth[v] - depth[u]) >> k & 1) {
v = parent[k][v];
}
}
if(u == v) return u;
for(int k = LOG_N - 1; k >= 0; k--)
{
if(parent[k][u] != parent[k][v]) {
u = parent[k][u];
v = parent[k][v];
}
}
return parent[0][u];
}
void Merge(int root)
{
if(tree[lson].depth > tree[rson].depth) tree[root].depth = tree[rson].depth, tree[root].id = tree[rson].id;
else tree[root].depth = tree[lson].depth, tree[root].id = tree[lson].id;
}
void build(int root, int l, int r)
{
if(l == r)
{
tree[root].id = lc[l];
tree[root].depth = depth[lc[l]];
return ;
} else if(r > l) {//r > l是为了防止n是1的时候,l就是1, r就是0了。
MID;
build(lson, l, mid);
build(rson, mid+1, r);
Merge(root);
}
}
int ansdepth, ansid;
void Query(int root, int l, int r, int ul, int ur)
{
if(ul <= l && r <= ur)
{
if(tree[root].depth < ansdepth)
ansdepth = tree[root].depth, ansid = tree[root].id;
return ;
}
MID;
if(mid >= ul) Query(lson, l, mid, ul, ur);
if(mid < ur) Query(rson, mid+1, r, ul, ur);
}
int main()
{
int n, u, v, q;
scanf("%d", &n);
for(int i = 1; i < n; i++)
{
scanf("%d %d", &u, &v);
G[u].push_back(v);
G[v].push_back(u);
}
init(n);
for(int i = 1; i < n; i++) lc[i] = lca(i, i+1);
build(1, 1, n-1);
scanf("%d", &q);
while(q--)
{
scanf("%d %d", &u, &v);
if(u > v) swap(u, v);
ansdepth = 0x3f3f3f3f, ansid = u;
if(u != v) Query(1, 1, n-1, u, v-1);
printf("%d\n", ansid);
}
}