codeforces 519E A and B and Lecture Rooms LCA倍增
Description
A and B are preparing themselves for programming contests.
The University where A and B study is a set of rooms connected by corridors. Overall, the University has n rooms connected by n - 1corridors so that you can get from any room to any other one by moving along the corridors. The rooms are numbered from 1 to n.
Every day А and B write contests in some rooms of their university, and after each contest they gather together in the same room and discuss problems. A and B want the distance from the rooms where problems are discussed to the rooms where contests are written to be equal. The distance between two rooms is the number of edges on the shortest path between them.
As they write contests in new rooms every day, they asked you to help them find the number of possible rooms to discuss problems for each of the following m days.
Input
The first line contains integer n (1 ≤ n ≤ 105) — the number of rooms in the University.
The next n - 1 lines describe the corridors. The i-th of these lines (1 ≤ i ≤ n - 1) contains two integers ai and bi (1 ≤ ai, bi ≤ n), showing that the i-th corridor connects rooms ai and bi.
The next line contains integer m (1 ≤ m ≤ 105) — the number of queries.
Next m lines describe the queries. The j-th of these lines (1 ≤ j ≤ m) contains two integers xj and yj (1 ≤ xj, yj ≤ n) that means that on the j-th day A will write the contest in the room xj, B will write in the room yj.
Output
In the i-th (1 ≤ i ≤ m) line print the number of rooms that are equidistant from the rooms where A and B write contest on the i-th day.
Sample Input
4
1 2
1 3
2 4
1
2 3
1
4
1 2
2 3
2 4
2
1 2
1 3
0
2
Hint
in the first sample there is only one room at the same distance from rooms number 2 and 3 — room number 1.
思路:
题意:给你一棵有n个节点的树,m个询问,每个询问a,b,求树中到a,b距离相等的节点个数
我们先使得deg[a] >= deg[b]
首先,若a~b之间的距离为奇数的话,找不到唯一的中点,此时答案为0
(怎么求a~b之间的距离呢?预处理节点i到根的距离(此处为深度)deg[i],求出lcaq = lca(a,b);
那么dist(a,b) = deg[a] - deg[lcaq] + deg[b] - deg[lcaq];)
否则:找出ab路径的中点mid,calc(u, d)可以找出节点u的深度为d的祖先,也就是我们求出中点的深度就可以找出中点!
怎么求中点的深度dmid?通过画图可知:
dmid = dist(a,b) / 2 - deg[b] + 2 * deg[lcaq];
最终答案的计算:num[i]表示以节点i为根的子树的节点总数
若a和b的深度相同,那么ans = num[lcaq] - num[xa] - num[xb]; 其中xa是lcaq在a~lcaq的下一个节点,xb是lcaq在b~lcaq的下一个节点
若a和b深度不同,那么ans = num[mid] - num[k], 其中k表示ab的中点的下一位置(在a和b路径上靠近a的下一位置),画图可以理解。
#include <bits/stdc++.h>
using namespace std; const int N = 1e5 + 5;
struct edge {
int v, to;
edge() {};
edge(int v, int to) : v(v), to(to) {};
}e[N << 1];
int head[N], num[N], deg[N], tot, n, m;
int p[N][30]; void init()
{
memset(head, -1, sizeof head);
memset(p, -1, sizeof p);
tot = 0;
}
void addedge(int u, int v)
{
e[tot] = edge(v, head[u]);
head[u] = tot++;
}
void dfs(int u, int fa, int d)
{
deg[u] = d; p[u][0] = fa; num[u] = 1;
for(int i = head[u]; ~i; i = e[i].to) {
int v = e[i].v;
if(v != fa) {
dfs(v, u, d + 1);
num[u] += num[v];
}
}
}
void pre()
{
for(int j = 0; (1 << j) <= n; ++j)
for(int i = 1; i <= n; ++i) {
if(p[i][j - 1] != -1)
p[i][j] = p[ p[i][j - 1] ][j - 1];
}
} int calc(int u, int d) ///返回节点u的深度为d的祖先
{
for(int j = 25; j >= 0; --j) {
if(deg[u] - (1 << j) >= d) u = p[u][j];
}
return u;
}
int lca(int a, int b)
{
a = calc(a, deg[b]);///使a和b处在同一层
if(a == b) return a; ///若此时a和b相等,那么lca就是a
///否则a和b同时向上爬
for(int j = 25; j >= 0; --j) {
if(p[a][j] != -1 && p[a][j] != p[b][j]) {
a = p[a][j];
b = p[b][j];
}
}
return p[a][0];
}
int main()
{
while(~scanf("%d", &n))
{
init();
int u, v;
for(int i = 1; i < n; ++i) {
scanf("%d%d", &u, &v);
addedge(u, v);
addedge(v, u);
}
dfs(1, -1, 0);
pre(); scanf("%d", &m);
int a, b;
while(m --)
{
scanf("%d%d", &a, &b);
if(deg[a] < deg[b]) swap(a, b);
if(a == b) { printf("%d\n", n); continue; }
int lcaq = lca(a, b);
int dist = deg[a] + deg[b] - 2 * deg[lcaq]; ///a和b之间的距离
if(dist & 1) { puts("0"); continue; }
if(deg[a] == deg[b]) { ///xa和xb分别是在a~lca和b~lca的路径上距离lca为1的点
int xa = calc(a, deg[lcaq] + 1);
int xb = calc(b, deg[lcaq] + 1);
printf("%d\n", n - num[xa] - num[xb]);
}
else {
int mid = dist / 2 - deg[b] + 2 * deg[lcaq]; ///mid为ab之间的路径的中点的深度
printf("%d\n", num[ calc(a, mid) ] - num[ calc(a, mid + 1) ]);
}
}
}
return 0;
}
codeforces 519E A and B and Lecture Rooms LCA倍增的更多相关文章
- codeforces 519E A and B and Lecture Rooms(LCA,倍增)
转载请注明出处: http://www.cnblogs.com/fraud/ ——by fraud E. A and B and Lecture Rooms A and B are ...
- Codeforces 519E A and B and Lecture Rooms
http://codeforces.com/contest/519/problem/E 题意: 给出一棵树和m次询问,每次询问给出两个点,求出到这两个点距离相等的点的个数. 思路: lca...然后直 ...
- CodeForces 519E A and B and Lecture Rooms(倍增)
A and B are preparing themselves for programming contests. The University where A and B study is a s ...
- Codeforces 519E A and B and Lecture Rooms [倍增法LCA]
题意: 给你一棵有n个节点的树,给你m次询问,查询给两个点,问树上有多少个点到这两个点的距离是相等的.树上所有边的边权是1. 思路: 很容易想到通过记录dep和找到lca来找到两个点之间的距离,然后分 ...
- [codeforces 519E]E. A and B and Lecture Rooms(树上倍增)
题目:http://codeforces.com/problemset/problem/519/E 题意:给你一个n个点的树,有m个询问(x,y),对于每个询问回答树上有多少个点和x,y点的距离相等 ...
- A and B and Lecture Rooms(LCA)
题目描述 A and B are preparing themselves for programming contests. The University where A and B study i ...
- Codeforces Round #294 (Div. 2) A and B and Lecture Rooms(LCA 倍增)
A and B and Lecture Rooms time limit per test 2 seconds memory limit per test 256 megabytes input st ...
- [CF Round #294 div2] E. A and B and Lecture Rooms 【树上倍增】
题目链接:E. A and B and Lecture Rooms 题目大意 给定一颗节点数10^5的树,有10^5个询问,每次询问树上到xi, yi这两个点距离相等的点有多少个. 题目分析 若 x= ...
- CodeForces 519E 树形DP A and B and Lecture Rooms
给出一棵树,有若干次询问,每次询问距两个点u, v距离相等的点的个数. 情况还挺多的,少侠不妨去看官方题解.^_^ #include <iostream> #include <cst ...
随机推荐
- 【leetcode】Plus One (easy)
Given a non-negative number represented as an array of digits, plus one to the number. The digits ar ...
- 【python】Head First Python(五)
无聊,看看<Head First Python>打发一下时间.感觉这本书很一般,可以无聊的时候翻翻.每一章页数很多,但都没讲什么东西. 先看第五章.记录一下知识点: f.readline( ...
- ios手势
iOS 手势操作:拖动.捏合.旋转.点按.长按.轻扫.自定义 大 中 小 1.UIGestureRecognizer 介绍 手势识别在 iOS 中非常重要,他极大地提高了移动设备的使用便捷性. i ...
- SQLServer基本操作
SQL 全名是结构化查询语言(Structured Query Language),是关系数据库管理系统的标准语言 1.分离数据库:将当前数据库文件和数据库引擎的关系断开,没有任何关系了,这样就可以随 ...
- Eclipse启动时出现错误 An internal error occurred during: “Updating indexes”
在Eclipse的workspace下有个.metadata文件夹,Eclipse出现异常的log文件就在这个目录下. 最近出现了这样的错误: 查看日志文件发现: !ENTRY org.ecl ...
- 通过btn获取所在cell
[cell.btn addTarget:self action:@selector(cellBtnClicked:event:) forControlEvents:UIControlEventTouc ...
- redis 常用命令
查看redis信息和状态: > info redis下,数据库是由一个整数索引标识,而不是由一个数据库名称.默认情况下,一个客户端连接到数据库0.redis配置文件中下面的参数来控制数据库总数: ...
- 如何在elasticsearch中查看Logstash打到elasticsearch的数据
# cat syslog02.conf #filename:syslog02.conf #注意这个是要用#号注释掉 input{ file{ path => ["/var/log/*. ...
- Shell编程基础教程7--脚本参数的传递
7.脚本参数的传递 7.1.shift命令 简介: shift n 每次将参数位置向左偏移n位 例子 #!/bin/bash us ...
- Pyqt 时时CPU使用情况
借鉴代码来自:https://github.com/hgoldfish/quickpanel 实现代码: # -*- coding:utf-8 -*- from __future__ import p ...