题意:
根据题意描述构建一棵树,树的节点可能达到20000000个,不过只有n (n <= 300000)个节点有儿子节点,每次询问x , y,问x是否为y的祖先。
解题思路:
因为只有n个节点有儿子,所以每次询问x、y时,x必须是有儿子的也就是说x < n,然后可以二分找到y的父亲,判断下x和y的父亲的 最近公共祖先是否是x即可,不过直接dfs构建lca会爆栈的,所以需要用栈模拟dfs。
模拟栈,每次取出栈头元素,但是先不要删掉,遍历该元素的儿子,如果儿子都被访问过了则删除栈头元素,否则加入一个儿子跳出,因为有回溯,其实再一次访问到这个元素就相当于回溯。注意每次加入一个儿子。具体见代码~
#include <stdio.h>
#include <algorithm>
using namespace std;
const int maxn = 300000 + 10;
struct Edge {
int to, next;
}edge[maxn];
int head[maxn], dfn[maxn], B[maxn<<2], F[maxn<<2], d[maxn<<2][20], pos[maxn<<2];
int E, Time, tot;
void newedge(int u, int to) {
edge[E].to = to;
edge[E].next = head[u];
head[u] = E++;
}
void init(int n) {
for(int i = 0;i <= n; i++) {
head[i] = -1;
dfn[i] = 0;
}
E = Time = tot = 0;
}
// 构建rmq && lca
void rmq_init(int n) {
for(int i = 1;i <= n; i++) d[i][0] = B[i];
for(int j = 1;(1<<j) <= n; j++)
for(int i = 1;i + j - 1 <= n; i++)
d[i][j] = min(d[i][j-1], d[i + (1<<(j-1))][j-1]);
}
int rmq(int L, int R) {
int k = 0;
while( (1<<(k+1)) <= R-L+1) k++;
return min(d[L][k], d[R-(1<<k)+1][k]);
}
int lca(int a, int b) {
if(pos[a] > pos[b]) swap(a, b);
int ans = rmq(pos[a], pos[b]);
return F[ans];
}
int st[maxn];
// 用栈模拟dfs
void dfs() {
int top = 0;
st[++top] = 0;
while(top) {
int u = st[top];
if(!dfn[u]) {
dfn[u] = ++Time;
F[Time] = u;
pos[u] = tot+1;
}
B[++tot] = dfn[u];
bool flag = 0;
for(int i = head[u];i != -1;i = edge[i].next) {
int to = edge[i].to;
if(!dfn[to]) {
flag = 1;
st[++top] = to;
break; // 加入一个儿子后马上跳出
}
}
if(!flag) top--;
}
}
int c[maxn], n;
// 二分查找父亲
int bin(int l, int r, int val) {
while(l < r) {
int mid = (l+r)/2;
if(c[mid] >= val)
r = mid;
else
l = mid+1;
}
return r;
}
void solve() {
dfs();
rmq_init(tot);
int m, u, to;
scanf("%d", &m);
while(m--) {
scanf("%d%d", &u, &to);
if(u >= n) {
puts("No"); continue;
}
if(u == to) {
puts("No"); continue;
}
int k = bin(0, n-1, to);
if(lca(u, k) == u) puts("Yes");
else puts("No");
}
}
int main() {
int t, m, cas = 1;
scanf("%d", &t);
while(t--) {
scanf("%d", &n);
init(n);
for(int i = 0;i < n; i++)
scanf("%d", &c[i]);
for(int i = 1;i < n; i++)
c[i] += c[i-1];
for(int i = 1;i < n; i++) {
int k = bin(0, n-1, i);
newedge(k, i);
}
printf("Case %d:\n", cas++);
solve();
if(t) puts("");
}
return 0;
}