Construct Binary Tree from Preorder and Inorder Traversal 두 갈래 트리 @LeetCode
package Level4;
import Utility.TreeNode;
/**
*
* Construct Binary Tree from Preorder and Inorder Traversal
*
* Given preorder and inorder traversal of a tree, construct the binary tree.
*
* Note: You may assume that duplicates do not exist in the tree.
*
*/
public class S140 {
public static void main(String[] args) {
int[] preorder = {1};
int[] inorder = {1};
buildTree(preorder, inorder);
}
public static TreeNode buildTree(int[] preorder, int[] inorder) {
int len = preorder.length;
if(len == 0){
return null;
}
return rec(preorder, inorder, 0, len, 0, len);
}
public static TreeNode rec(int[] preorder, int[] inorder, int preStart, int preEnd, int inStart, int inEnd){
if(preStart >= preorder.length){ // !
return null;
}
TreeNode root = new TreeNode(preorder[preStart]);
int rootIndex;
for(rootIndex=inStart; rootIndex<=inEnd; rootIndex++) {
if(inorder[rootIndex] == preorder[preStart]){ // inorder root
break;
}
}
int len = rootIndex - inStart; //
if(rootIndex > inStart){
root.left = rec(preorder, inorder, preStart+1, preStart+len, inStart, rootIndex-1);
}
if(rootIndex < inEnd){
root.right = rec(preorder, inorder, preStart+len+1, preEnd, rootIndex+1, inEnd);
}
return root;
}
}
다음은 또 다른 쓰기 방법으로 preStart의 변화를 피하지만, 단점은 전역 변수로 preIndex를 기록하는 것이다
int preIndex = 0;
public TreeNode buildTree(int[] preorder, int[] inorder) {
return rec(preorder, inorder, 0, inorder.length-1);
}
public TreeNode rec(int[] preorder, int[] inorder, int inStart, int inEnd) {
if (inStart > inEnd) {
return null;
}
int rootData = preorder[preIndex++];
TreeNode root = new TreeNode(rootData);
int inIndex = search(inorder, inStart, inEnd, rootData);
root.left = rec(preorder, inorder, inStart, inIndex-1);
root.right = rec(preorder, inorder, inIndex+1, inEnd);
return root;
}
private int search(int[] inorder, int inStart, int inEnd, int data) {
for(int i=inStart; i<=inEnd; i++) {
if(inorder[i] == data) {
return i;
}
}
return -1;
}
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public TreeNode buildTree(int[] preorder, int[] inorder) {
return rec(preorder, inorder, 0, preorder.length-1, 0, inorder.length-1);
}
public TreeNode rec(int[] preorder, int[] inorder, int preorderLeft, int preorderRight, int inorderLeft, int inorderRight) {
if(preorderLeft > preorderRight) {
return null;
}
TreeNode root = new TreeNode(preorder[preorderLeft]);
int inorderRootPos = 0;
for(; inorderRootPos < inorderRight; inorderRootPos++) {
if(inorder[inorderRootPos] == root.val) {
break;
}
}
int dist = inorderRootPos - inorderLeft;
root.left = rec(preorder, inorder, preorderLeft+1, preorderLeft+dist, inorderLeft, inorderRootPos-1);
root.right = rec(preorder, inorder, preorderLeft+dist+1, preorderRight, inorderRootPos+1, inorderRight);
return root;
}
}
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
다양한 언어의 JSONJSON은 Javascript 표기법을 사용하여 데이터 구조를 레이아웃하는 데이터 형식입니다. 그러나 Javascript가 코드에서 이러한 구조를 나타낼 수 있는 유일한 언어는 아닙니다. 저는 일반적으로 '객체'{}...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.