码迷,mamicode.com
首页 > 其他好文 > 详细

【剑指offer】树的子结构

时间:2014-05-18 14:09:41      阅读:333      评论:0      收藏:0      [点我收藏+]

标签:剑指offer   树的子结构   

转载请注明出处:http://blog.csdn.net/ns_code/article/details/25907685


    剑指offer第18题,九度OJ上测试通过!

题目描述:

输入两颗二叉树A,B,判断B是不是A的子结构。

输入:

输入可能包含多个测试样例,输入以EOF结束。
对于每个测试案例,输入的第一行一个整数n,m(1<=n<=1000,1<=m<=1000):n代表将要输入的二叉树A的节点个数(节点从1开始计数),m代表将要输入的二叉树B的节点个数(节点从1开始计数)。接下来一行有n个数,每个数代表A树中第i个元素的数值,接下来有n行,第一个数Ki代表第i个节点的子孩子个数,接下来有Ki个树,代表节点i子孩子节点标号。接下来m+1行,与树A描述相同。

输出:

对应每个测试案例,
若B是A的子树输出”YES”(不包含引号)。否则,输出“NO”(不包含引号)。

样例输入:
7 3
8 8 7 9 2 4 7
2 2 3
2 4 5
0
0
2 6 7
0
0
8 9 2
2 2 3
0
0

1 1
2
0
3
0
样例输出:
YES
NO
提示:

B为空树时不是任何树的子树

    在写这道题目时,卡在了测试代码上,这个题目的测试代码有点繁杂,最后参考了一哥的文章,改用数组作为存储二叉树节点的数据结构,果然写测试代码方便了很多。

    另外,程序中有一些要注意的地方,在程序中表明了注释。

    AC代码:

#include<stdio.h>
#include<stdlib.h>
#include<stdbool.h>

typedef struct BTNode
{
	int data;
	int rchild;
	int lchild;
}BTNode;

/*
判断pTree2是否是与pTree1有共同的根节点的pTree1子树
*/
bool isSubTree(BTNode *pTree1,int index1,BTNode *pTree2,int index2)
{
	//前两个if语句不能颠倒,不然当pTree1和pTree2相同时,会误判为false
	if(index2 == -1)
		return true;
	if(index1 == -1)
		return false;
	if(pTree1[index1].data != pTree2[index2].data)
		return false;
	else
		return isSubTree(pTree1,pTree1[index1].lchild,pTree2,pTree2[index2].lchild) &&
			   isSubTree(pTree1,pTree1[index1].rchild,pTree2,pTree2[index2].rchild);
}

/*
判断pTree1是否包含pTree2
*/
bool isContainTree(BTNode *pTree1,int index1,BTNode *pTree2,int index2)
{
	if(pTree1==NULL || pTree2==NULL)
		return false;
	if(index1==-1 || index2==-1)
		return false;

	bool result = false;
	if(pTree1[index1].data == pTree2[index2].data)
		result = isSubTree(pTree1,index1,pTree2,index2);

	//如果pTree1[index1].lchild为-1,下次递归时会通过index1==-1的判断返回false,
	//因此这里不需要再加上pTree1[index1].lchild!=-1的判断条件
	if(!result)
		result = isContainTree(pTree1,pTree1[index1].lchild,pTree2,index2);
	if(!result)
		result = isContainTree(pTree1,pTree1[index1].rchild,pTree2,index2);

	return result;
}

int main()
{
	int n,m;
	while(scanf("%d %d",&n,&m) != EOF)
	{
		//输入树pTree1各节点的值
		BTNode *pTree1 = NULL;
		if(n>0)
		{
			pTree1 = (BTNode *)malloc(n*sizeof(BTNode));
			if(pTree1 == NULL)
				exit(EXIT_FAILURE);
			int i,data;
			//输入n个节点的data
			for(i=0;i<n;i++)
			{
				scanf("%d",&data);
				pTree1[i].data = data;
				pTree1[i].rchild = -1;
				pTree1[i].lchild = -1;
			}

			//输入n行节点连接关系
			for(i=0;i<n;i++)
			{
				int ki;
				scanf("%d",&ki);
				if(ki == 0)
					continue;
				else if(ki == 1)
				{
					int lindex;
					scanf("%d",&lindex);
					pTree1[i].lchild = lindex-1;
				}
				else
				{
					int lindex,rindex;
					scanf("%d",&lindex);
					scanf("%d",&rindex);
					pTree1[i].lchild = lindex-1;
					pTree1[i].rchild = rindex-1;
				}
			}
		}

		//输入树pTree2各节点的值
		BTNode *pTree2 = NULL;
		if(m>0)
		{
			pTree2 = (BTNode *)malloc(m*sizeof(BTNode));
			if(pTree2 == NULL)
				exit(EXIT_FAILURE);
			int i,data;
			//输入n个节点的data
			for(i=0;i<m;i++)
			{
				scanf("%d",&data);
				pTree2[i].data = data;
				pTree2[i].rchild = -1;
				pTree2[i].lchild = -1;
			}

			//输入n行节点连接关系
			for(i=0;i<m;i++)
			{
				int ki;
				scanf("%d",&ki);
				if(ki == 0)
					continue;
				else if(ki == 1)
				{
					int lindex;
					scanf("%d",&lindex);
					pTree2[i].lchild = lindex-1;
				}
				else
				{
					int lindex,rindex;
					scanf("%d",&lindex);
					scanf("%d",&rindex);
					pTree2[i].lchild = lindex-1;
					pTree2[i].rchild = rindex-1;
				}
			}
		}

		if(isContainTree(pTree1,0,pTree2,0))
			printf("YES\n");
		else
			printf("NO\n");

	}
	return 0;
}

/**************************************************************
    Problem: 1520
    User: mmc_maodun
    Language: C
    Result: Accepted
    Time:10 ms
    Memory:912 kb
****************************************************************/


【剑指offer】树的子结构,布布扣,bubuko.com

【剑指offer】树的子结构

标签:剑指offer   树的子结构   

原文地址:http://blog.csdn.net/ns_code/article/details/25907685

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!