root1表示树1的根节点,root2表示树2的根节点。判断树2是否是树1的子树,是返回1,不是返回-1。子树不同于树的一部分。
#include<iostream>
using namespace std;
struct tnode{
int value;
tnode*left;
tnode*right;
};
//深度优先遍历 在树中寻找节点node 层次遍历更好
tnode * find(tnode* root, tnode* node){
if(root == NULL || node == NULL) return NULL;
if(root->value == node->value) return root;
tnode* tmp=find(root->left,node);
if(tmp) return tmp;
return find(root->right,node);
}
//判断两个树是否相等
bool same(tnode* node1, tnode* node2){
if(node1==NULL && node2==NULL) return 1;
if(node1==NULL && node2!=NULL) return 0;
if(node1!=NULL && node2==NULL) return 0;
if(node1->value == node2->value)
return same(node1->left,node2->left) && same(node1->right,node2->right);
return 0;
}
int isSubTree(tnode* root1, tnode* root2)
{
if(root1==NULL || root2==NULL) return -1;
//首先遍历root1,找到root2对应的节点
tnode* tmp=find(root1,root2);
if(tmp==NULL) return -1;//root2不在root1中
//接下来判断tmp和root2为根的子树是否相等
//cout<<tmp->value<<" "<<root2->value<<endl;
if(same(tmp,root2)) return 1;
else return -1;
}
//主函数用于测试
int main(){
tnode *a=new tnode[7];;
a[0].value=4;
a[1].value=2;
a[2].value=6;
a[3].value=1;
a[4].value=3;
a[5].value=5;
a[6].value=7;
for(int i=0;i<3;i++){
a[i].left=&a[2*i+1];
a[i].right=&a[2*i+2];
//cout<<a[i].value<<" "<<a[i].left->value<<" "<<a[i].right->value<<endl;
}
for(int i=3;i<7;i++){
a[i].left=NULL;
a[i].right=NULL;
}
tnode *b=new tnode[7];;
b[0].value=6;
b[1].value=5;
b[2].value=7;
b[0].left=&b[1];
b[0].right=&b[2];
b[1].left=NULL;
b[1].right=NULL;
b[2].left=NULL;
b[2].right=NULL;
cout<<isSubTree(&a[0],&b[0])<<endl;
system("pause");
}