두 갈래 나무의 중차적 역주행과 후차적 역주행 서열을 이용하여 두 갈래 나무 만들기 Search results for Construct Binary Tree from Inorder and Postorder Traversa

2461 단어
먼저 가장 간단한 세 가지 원소의 상황을 살펴보자.
inorder의 순서는 다음과 같습니다. B A C
postorde를 뒤따르는 순서: B C A
이 두 갈래 나무를 만들기 위해서 우리는 세 걸음으로 나누어 간다.
(1) 루트 결점을 찾습니다.
뒷순서를 이용하여 서열을 두루 훑어보다.후속 서열에서 우리는 이 두 갈래 나무의 루트가 A라는 것을 쉽게 얻을 수 있다.
(2) 루트의 왼쪽 나무를 찾습니다.
중서를 이용하여 서열을 두루 훑어보다.중간 시퀀스에서 루트 결점 A를 찾으면 A의 왼쪽 시퀀스 B가 루트의 왼쪽 트리입니다.
(3) 루트의 오른쪽 나무를 찾아라.
마찬가지로 중서를 이용하여 서열을 두루 훑어본다.중간 시퀀스에서 루트 결점 A를 찾으면 A의 오른쪽 시퀀스 C가 루트의 오른쪽 트리입니다.
이것은 간단한 세 가지 원소의 상황으로 상황이 복잡할 때 우리는 위의 세 가지 규칙으로 귀속 구조를 하면 된다.
아래의 테스트 코드는 참고할 수 있습니다
#include <vector>
#include <algorithm>
#include <iostream>
using namespace std;
/**
 * definite a binarytree
 *
 *
 */
struct TreeNode
{
	int val;
	TreeNode *left, *right;
	TreeNode(int x) :val(x), left(NULL), right(NULL){}
};

class Solution
{
public:
	Solution();
	~Solution();

	TreeNode *buildTree(vector<int> &inorder, vector<int> &postorder);
private:

};

Solution::Solution()
{
}

Solution::~Solution()
{
}

TreeNode *Solution::buildTree(vector<int> &inorder, vector<int> &postorder)
{
	
	TreeNode *root = new TreeNode(postorder.back());
	
	if (inorder.size() == 0 || postorder.size() == 0)
	{
		return NULL;
	}
	if (inorder.size() == 1)
	{
		return root;
	}

	

	vector<int>::iterator a = find(begin(inorder), end(inorder), postorder.back());// find() return an iterator
	vector<int> left(inorder.begin(), a);// 
	vector<int> right(a + 1, inorder.end());// 

	vector<int>::iterator b = find(postorder.begin(), postorder.end(), right.front());
	vector<int> left_p(postorder.begin(), b); 
	vector<int> right_p(b, postorder.end()-1); 



	root->left = buildTree(left,left_p);// 
	root->right = buildTree(right,right_p);// 

	return root;
}

template<typename T>
std::ostream &operator<<(std::ostream &s, const std::vector<T> &v)
{
	for (const auto e : v)
	{
		s << e << " ";
	}
	return s;
}


void printTreeBypreorder(const TreeNode *root)
{
	if (root)
	{
		cout << root->val << " ";
		printTreeBypreorder(root->left);
		printTreeBypreorder(root->right);
	}
	else
	{
		return;
	}
}
int main()
{


	vector<int> inorder{ 4, 2, 5, 1, 6, 3, 7 };
	vector<int> postorder{ 4, 5, 2, 6, 7, 3, 1 };
	Solution solu;
	TreeNode *result = solu.buildTree(inorder, postorder);
	printTreeBypreorder(result);
	return 0;
}

좋은 웹페이지 즐겨찾기