zoj2615-Cells

超时= =,可以用数组的存储结构来计算出子父节点,应该不会超时,懒得写了...

#include<iostream>
#include<queue>
#include<stack>
#include<stdio.h>
using namespace std;

const int chnum=101;

struct node
{
	int data;
	node *ch[chnum];
	node()
	{
		for(int i=0;i<chnum;i++)
			ch[i]=NULL;
	}
};

void del(node *&head)
{
	stack<node *> sta;
	node *p;
	sta.push(head);
	while(!sta.empty())
	{
		p=sta.top();
		sta.pop();
		for(int i=0;i<chnum;i++)
		{
			if(p->ch[i])
			{
				sta.push(p->ch[i]);
			}
		}
		delete p;
	}
}
void answer(node *,int,int);
int main()
{
	queue<node *> que;
	int icase,o=1;
	bool fl=false;
	scanf("%d",&icase);
	while(icase--)
	{
		if(fl)
			printf("\n");
		fl=true;
		while(!que.empty())
			que.pop();
		int num,k=0;
		scanf("%d",&num);
		node *p=NULL,*now=NULL,*head=NULL;
		p=new node;
		head=p;//头结点
		p->data=k++;
		que.push(p);
		while(num--)
		{
			int nonum;
			scanf("%d",&nonum);
			now=que.front();
			que.pop();
			for(int i=0;i<nonum;i++)
			{
				p=new node;
				now->ch[i]=p;
				p->data=k++;
				que.push(p);
			}
		}
		int testnum;
		scanf("%d",&testnum);
		bool flag= true;
		while(testnum--)
		{
			int a,b;
			scanf("%d%d",&a,&b);
			if(flag)
				printf("Case %d:\n",o++);
			flag=false;
			answer(head,a,b);
		}
		del(head);
	}
	return 0;
}
bool fine(node *&aima,node *anode,int num)//爆栈了= =,被迫改,还是超时..
{
	stack<node *> sta;
	node *p;
	sta.push(anode);
	while(!sta.empty())
	{
		p=sta.top();
		sta.pop();
		if(p->data==num)
		{
			aima=p;
			return true;
		}
		for(int i=0;i<chnum;i++)
		{
			if(p->ch[i])
			{
				sta.push(p->ch[i]);
			}
		}
	}
}
void answer(node *head,int a,int b)
{
	node *aim=NULL,*temp=NULL;
	fine(aim,head,a);
	fine(temp,aim,b);
	if(temp!=NULL)
		printf("Yes\n");
	else
		printf("No\n");
}