LibreOJ 10135：祖孙询问 ← DFS序 & 欧拉序+LCA+链式前向星

【题目来源】
https://loj.ac/p/10135
https://www.acwing.com/problem/content/1174/

【题目描述】
已知一棵 n 个节点的有根树。有 m 个询问，每个询问给出了一对节点的编号 x 和 y，询问 x 与 y 的祖孙关系。

【输入格式】
输入第一行包括一个整数 n 表示节点个数；
接下来 n 行每行一对整数对 a 和 b 表示 a 和 b 之间有连边。如果 b 是 -1，那么 a 就是树的根；
第 n+2 行是一个整数 m 表示询问个数；
接下来 m 行，每行两个正整数 x 和 y，表示一个询问。

【输出格式】
对于每一个询问，若 x 是 y 的祖先则输出 1，若 y 是 x 的祖先则输出 2，否则输出 0。

【输入样例】
10
234 -1
12 234
13 234
14 234
15 234
16 234
17 234
18 234
19 234
233 19
5
234 233
233 12
233 13
233 15
233 19

【输出样例】
1
0
0
0
2

【数据范围】
对于 30% 的数据，1≤n,m≤10^3；
对于 100% 的数据，1≤n,m≤4×10^4，每个节点的编号都不超过 4×10^4。

【算法分析】
● DFS 序
DFS 序表示从根结点开始对树进行 DFS 所得的结点遍历顺序。

易得上图的 DFS 序为：1，2，3，4，5，6，7，8，9。可见，通过 DFS 序，可将一棵树映射为一个一维数组。
假设以某结点 u 为根的子树大小为 cnt[u]，u 在整棵树中的 DFS 序为 dfs[u]，则可得结点 u 的所有子树对应的 DFS 序区间为 [dfs[u],dfs[u]+cnt[u]-1]。 → 这条性质是编写本题代码的关键。
容易发现，一棵子树所有结点的 DFS 序是整棵树的 DFS 序的连续一段。例如，上图中以 d 为根的子树，它的各结点的 DFS 序 3,4,5,6 是整棵树的 DFS 序 1，2，3，4，5，6，7，8，9 中的连续一段。可见，借助DFS序，可以快速判断一个结点是否在某子树内。对某子树进行操作，相当于在对应的 DFS 序区间内进行操作。
利用链式前向星存图，求树的 DFS 序的代码如下：

void dfs_seq(int u,int v) { //dfs sequence
    dfs[u]=++id;
    cnt[u]=1;
    for(int i=h[u]; ~i; i=ne[i]) {
        int j=e[i];
        if(j==v) continue;
        dfs_seq(j,u);
        cnt[u]+=cnt[j];
    }
    return ;
}

● 欧拉序
欧拉序与 DFS 序很像，但也有不同。欧拉序表示从根结点出发，按照 DFS 经过所有结点并返回根结点所得的结点遍历顺序（即返回时也要记录）。
若设 first[u] 是欧拉序中某结点 u 第一次出现的位置，last[u] 是欧拉序中 u 最后一次出现的位置，则以结点 u 为根的子树包含的结点，是欧拉序区间 [first[u], last[u]] 之间的所有结点。换句话说，就是 [first[u], last[u]] 之间的结点为 u 的子结点。
欧拉序有两种情况：
（1）DFS 时，第一次访问某结点时记录一次，随后每访问完该结点的一棵子树就再记录一次，一共有 2n-1 个编号。其中，n 是树中结点的个数。

易得上图的欧拉序为：1，2，3，4，3，5，3，6，3，2，7，2，1，8，9，8，1。这是第一种情况下的欧拉序。
针对第一种情况的欧拉序，具有如下性质：即若设 first[u] 是欧拉序中某结点 u 第一次出现的位置，first[v] 是欧拉序中某结点 v 第一次出现的位置，树上两结点 u, v 的最近公共祖先（LCA），为欧拉序区间 [first[u], first[v]] 或 [first[v], first[u]] 中时间戳最小的结点。其中，某结点的时间戳可以理解为第一次 DFS 遍历到该结点的顺序。  ------→ 再次特别提醒，此性质仅适用于第一种情况下的欧拉序。
利用链式前向星存图，在第一种情况下，求树的欧拉序的代码如下：

void ola_seq1(int u,int v) { //ola sequence 1
    dfs[++id]=u;
    cnt[u]=1;
    for(int i=h[u]; ~i; i=ne[i]) {
        int j=e[i];
        if(j==v) continue;
        ola_seq1(j,u);
        dfs[++id]=u;
        cnt[u]+=cnt[j];
    }
    return ;
}

（2）DFS 时，每个结点入栈与出栈时分别记录一次，共 2n 个编号。其中，n 是树中结点的个数。

易得上图的欧拉序为：1，2，3，4，4，5，5，6，6，3，7，7，2，8，9，9，8，1。这是第二种情况下的欧拉序。
利用链式前向星存图，在第二种情况下，求树的欧拉序的代码如下：

void ola_seq2(int u,int v) { //ola sequence 2
    dfs[++id]=u;
    cnt[u]=1;
    for(int i=h[u]; ~i; i=ne[i]) {
        int j=e[i];
        if(j==v) continue;
        ola_seq2(j,u);        
        cnt[u]+=cnt[j];
    }
    dfs[++id]=u;
    return;
}

● 链式前向星：https://blog.csdn.net/hnjzsyjyj/article/details/139369904
val[idx]：存储编号为 idx 的边的值
e[idx]：存储编号为 idx 的结点的值
ne[idx]：存储编号为 idx 的结点指向的结点的编号
h[a]：存储头结点 a 指向的结点的编号

【算法代码】

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

const int N=4e4+5;
const int M=N<<1;
int root;
int dfs[N],cnt[N],id;
int h[N],e[M],ne[M],idx;
int n,m;

void add(int a,int b) {
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

void dfs_seq(int u,int v) { //dfs sequence
    dfs[u]=++id;
    cnt[u]=1;
    for(int i=h[u]; ~i; i=ne[i]) {
        int j=e[i];
        if(j==v) continue;
        dfs_seq(j,u);
        cnt[u]+=cnt[j];
    }
    return ;
}

int main() {
    cin>>n;
    memset(h,-1,sizeof(h));
    int a,b;
    for(int i=1; i<=n; i++) {
        cin>>a>>b;
        if(b==-1) root=a;
        else add(a,b),add(b,a);
    }
    dfs_seq(root,-1);

    cin>>m;
    for(int i=1; i<=m; i++) {
        cin>>a>>b;
        if(dfs[a]<dfs[b] && dfs[a]+cnt[a]-1>=dfs[b]) cout<<"1"<<endl;
        else if(dfs[b]<dfs[a] && dfs[b]+cnt[b]-1>=dfs[a]) cout<<"2"<<endl;
        else cout<<"0"<<endl;
    }

    return 0;
}

/*
in:
10
234 -1
12 234
13 234
14 234
15 234
16 234
17 234
18 234
19 234
233 19
5
234 233
233 12
233 13
233 15
233 19

out:
1
0
0
0
2
*/

【参考文献】
https://www.cnblogs.com/hetailang/p/16209936.html
https://blog.csdn.net/littlegengjie/article/details/134430600
https://blog.csdn.net/zht2002/article/details/128686361