ZOJ 2615——用栈模拟dfs

ZOJ 2615整整弄了两天，虽然说题目意思很简单，就是dfs。但是由于数据过大，不得不采取用栈去模拟dfs的方法，下面做一下总结

题目链接：http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=13618

题目大意就是给你两个数，询问前一个数是否为后一个数的祖先，在做题过程中，由于查询次数过多，所以借鉴了网上的思路，用出入栈次序去判断。

代码如下：

//基本思路：在模拟dfs过程中子节点的进栈一定晚于父节点，出栈一定早于父节点
#include<stdio.h>
#define MAXM 20000005
#define MAXN 300005
#pragma comment(linker,"/STACK:1024000000,1024000000")
int num[MAXN],child[MAXN],now[MAXN],n,m,top,my_stack[MAXM],top_stack,in[MAXM],out[MAXM];//now用来记录当前判断的子节点，in，out用来记录进栈出栈的次序
void my_dfs()//用栈模拟dfs
{
    for(int i=0;i<n;i++)
        now[i]=0;
    int c=0;
    top_stack=0;
    my_stack[0]=0;
    in[0]=c++;
    int i;
    while(1)
    {
        if(my_stack[top_stack]>=n)//大于等于n一定没有子节点，直接出栈
        {
            out[my_stack[top_stack]]=c++;
            top_stack--;
        }
        i=child[my_stack[top_stack]]+now[my_stack[top_stack]];
        if(now[my_stack[top_stack]]==num[my_stack[top_stack]]&&top_stack==0)//判断是否遍历完
        {
            out[my_stack[top_stack]]=c;
            return ;
        }
        else if(now[my_stack[top_stack]]==num[my_stack[top_stack]])//判断当前栈元素的子节点是否遍历完想，遍历完就出栈
        {
            out[my_stack[top_stack]]=c++;
            top_stack--;
        }
        else//入栈
        {
            now[my_stack[top_stack]]++;
            my_stack[++top_stack]=i;
            in[i]=c++;
        }
    }
}
bool Query(int a,int b)
{
    if(a>=b||a>=n||in[a]>in[b]||out[a]<out[b])
        return false;
    return true;
}
int main()
{
    int t;
    scanf("%d",&t);
    for(int i=1;i<=t;i++)
    {
        if(i!=1)
            printf("\n");
        printf("Case %d:\n",i);
        scanf("%d",&n);
        top=1;
        for(int j=0;j<n;j++)
        {
            int a;
            scanf("%d",&a);
            num[j]=a;
            child[j]=top;
            top+=a;
        }
        my_dfs();
        scanf("%d",&m);
        for(int j=0;j<m;j++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            if(Query(a,b))
                printf("Yes\n");
            else
                printf("No\n");
        }
    }
    return 0;
}