면접 문제 20: 나무의 하위 구조(offer)

1632 단어
제목:
두 그루의 두 갈래 나무 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나무가 비교적 다르다면 다음 같은 결점을 계속 찾고 계속 비교해야 합니다.

좋은 웹페이지 즐겨찾기