LA 3486 Cells（判祖先+栈模拟dfs）

https://vjudge.net/problem/UVALive-3486

题意：

判断u是否是v的祖先。

思路：

很简单，dfs遍历，记录每个节点第一次访问时的时间戳 in[i] 和第二次访问时的时间戳 out[i]，如果满足in[x]<in[y]<out[x]，那么x就是y的祖先。

用dfs实现是很容易的事，但这题需要用栈来模拟一下。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<sstream>
#include<vector>
#include<stack>
#include<queue>
#include<cmath>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
typedef pair<int,int> pll;
const int INF = 0x3f3f3f3f;
const int maxn=20000000+5;

int n, m;
int start[maxn];
int c[maxn];
int in[maxn];
int out[maxn];

void dfs(int u)
{
    stack<int> S;
    int dfs_clock=0;
    in[u]=0;
    S.push(u);
    while(!S.empty())
    {
        int x=S.top();
        if(in[x])
        {
            out[x]=++dfs_clock;
            S.pop();
            continue;
        }
        in[x]=++dfs_clock;
        for(int i=start[x];i<start[x]+c[x];i++)
        {
            if(i<n)  {in[i]=0;S.push(i);}
            else  {in[i]=++dfs_clock;out[i]=++dfs_clock;}
        }
    }
}

int main()
{
    //freopen("in.txt","r",stdin);
    int T; int kase=0;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        int s=1;
        for(int i=0;i<n;i++)
        {
            scanf("%d",&c[i]);
            start[i]=s; s+=c[i];
        }
        dfs(0);
        printf("Case %d:
",++kase);
        int q;
        scanf("%d",&q);
        while(q--)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            if(in[u]<in[v] && out[u]>in[v])  puts("Yes");
            else puts("No");
        }
        if(T) puts("");
    }
    return 0;
}