Construct Binary Tree from Preorder and Inorder Traversal 두 갈래 트리 @LeetCode

3580 단어
고전적인 제목과 실제에서도 용도가 있다. 예를 들어 어떻게 두 갈래 나무를 서열화한 후에 인터넷을 통해 다른 기계에 전달하면 우리는 두 갈래 나무를preorder와inorder의 방식으로string으로 바꾸어 다른 기계에 전달하고 다른 기계는 두 갈래 나무를 재건할 수 있다.
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;
    }
}

좋은 웹페이지 즐겨찾기