2024/9/01 LeetCode Hot100 binaryTree

This commit is contained in:
Cool 2024-09-01 23:30:59 +08:00
parent 2dda42f570
commit 19151f857e
3 changed files with 126 additions and 1 deletions

View File

@ -0,0 +1,57 @@
package com.cool.hot100.binary_tree;
import org.junit.Test;
import java.util.HashMap;
import java.util.Map;
/**
* Created with IntelliJ IDEA.
*
* @Author: Cool
* @Date: 2024/09/01/19:34
* DayNumber 2
* Hard 2
* Level 5
*/
public class Num105 {
Map<Integer, Integer> map = new HashMap<>();
public TreeNode buildTree(int[] preorder, int[] inorder) {
for (int i = 0; i < inorder.length; i++) {
map.put(inorder[i], i);
}
return buildTree(preorder, 0, 0, inorder.length - 1);
}
public TreeNode buildTree(int[] preorder, int indexInPre, int startInIn, int endInIn) {
if (startInIn > endInIn) return null;
TreeNode node = new TreeNode(preorder[indexInPre]);
Integer headInIn = map.get(preorder[indexInPre]);
int leftLength = headInIn - startInIn;
int rightLength = endInIn - headInIn;
node.left = buildTree(preorder,indexInPre+1,startInIn,headInIn-1);
node.right=buildTree(preorder,indexInPre+leftLength+1,headInIn+1,endInIn);
return node;
}
@Test
public void test() {
// TreeNode node1 = new TreeNode(6);
// TreeNode node2 = new TreeNode(2);
// TreeNode node3 = new TreeNode(4);
// TreeNode node4 = new TreeNode(3);
// TreeNode node5 = new TreeNode(5);
// TreeNode node6 = new TreeNode(2);
// TreeNode node7 = new TreeNode(1);
// node7.left = node6;
// node7.right = node5;
// node6.left = node4;
// node6.right = node3;
// node5.right = node1;
buildTree(new int[]{3, 9, 20, 15, 7}, new int[]{9, 3, 15, 20, 7});
}
}

View File

@ -0,0 +1,69 @@
package com.cool.hot100.binary_tree;
import org.junit.Test;
/**
* Created with IntelliJ IDEA.
*
* @Author: Cool
* @Date: 2024/09/01/17:17
* DayNumber 1
* Hard 2
* Level 3
*/
public class Num114 {
public void flatten(TreeNode root) {
// while (root != null) {
// TreeNode right = root.right;
// if (root.left != null) {
// TreeNode left = root.left;
// while (left.right != null) {
// left = left.right;
// }
// root.right = root.left;
// left.right = right;
// }
// root.left = null;
// root = root.right;
// }
dfs(root);
}
private TreeNode dfs(TreeNode root) {
if (root == null) return null;
TreeNode returnRoot = root;
TreeNode right = dfs(root.right);
TreeNode left = dfs(root.left);
root.left = null;
if (left != null) {
root.right = left;
}
if (left != null && right != null) {
while (root.right!=null){
root=root.right;
}
root.right = right;
} else if (right != null) {
root.right = right;
}
return returnRoot;
}
@Test
public void test() {
TreeNode node1 = new TreeNode(6);
TreeNode node2 = new TreeNode(2);
TreeNode node3 = new TreeNode(4);
TreeNode node4 = new TreeNode(3);
TreeNode node5 = new TreeNode(5);
TreeNode node6 = new TreeNode(2);
TreeNode node7 = new TreeNode(1);
node7.left = node6;
node7.right = node5;
node6.left = node4;
node6.right = node3;
node5.right = node1;
flatten(node7);
}
}

View File

@ -1,6 +1,5 @@
package com.cool.hot100.binary_tree;
import sun.reflect.generics.tree.Tree;
import java.util.ArrayList;
import java.util.LinkedList;