UVa 1357 Cells (栈模拟dfs LCA)

题意：

根据题意描述构建一棵树，树的节点可能达到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;
}