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;
}