题目:
输入两棵二叉树A和B,判断B是不是A的子结构。二叉树结点定义如下:
struct BinaryTreeNode{
int value;
BinaryTreeNode *left;
BinaryTreeNode *right;
};
边界条件:
B为空,返回真。
A为空,返回假。
思路:1.先找到B的头结点在A中的位置,如果没找到,返回假。
2.找到位置后递归进行判断。
时间复杂度:O(lgn1)
#include <iostream>
#include <vector>
#include <string>
#include <stack>
#include <algorithm>
using namespace std;
struct BinaryTreeNode{
int value;
BinaryTreeNode *left;
BinaryTreeNode *right;
};
bool isSubTreeFromRoot(BinaryTreeNode *root1, BinaryTreeNode *root2)
{
if (root2 == NULL) return true;
if (root1 == NULL) return false;
if (root1->value != root2->value) return false;
return isSubTreeFromRoot(root1->left, root2->left) && isSubTreeFromRoot(root1->right, root2->right);
}
BinaryTreeNode *findNode(BinaryTreeNode *root, BinaryTreeNode *toBeFind)
{
if (root == NULL) return NULL;
if (root->value == toBeFind->value) return root;
if (root->value < toBeFind->value) return findNode(root->right, toBeFind);
else return findNode(root->left, toBeFind);
}
bool isSubTree(BinaryTreeNode *root1, BinaryTreeNode *root2)
{
if (root2 == NULL) return true;
if (root1 == NULL) return false;
BinaryTreeNode *pos = findNode(root1, root2);
return isSubTreeFromRoot(pos, root2);
}
int main()
{
return 0;
}
可惜,上面的解法,我把树当做二叉搜索树来做的,如果是二叉树的话,思路差不多,但是如果找到的结点和B树比较不同后,还要继续找下一个相同的结点,再继续比较。