树的学习缘由
leetcode每次周赛基本都会有一道树的题目,一直对这块的学习没那么深入,一般做出来的题目都是用递归去做,没有感受到树的优缺点。
这篇文章就作为树的深入学习的起点,之后各种树都尽量去了解和学习他们的代码写法和优缺点。
树的构建
因为目前树的使用基本知识leetcode上的做题,所以这次以leetcode上根据输入内容构建一颗树来开始树的学习。
以后写树的题目这块就可以做为基础类来本地调试。
package com.phw.demo;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
public class TreeNode {
int val;
TreeNode left;
TreeNode right;
Integer[] rootVal;
TreeNode(int x){
this.val=x;
}
public static TreeNode stringToTreeNode(String input) {
input = input.trim();
input = input.substring(1, input.length() - 1);
if (input.length() == 0) {
return null;
}
String[] parts = input.split(",");
String item = parts[0];
TreeNode root = new TreeNode(Integer.parseInt(item));
Queue<TreeNode> nodeQueue = new LinkedList<>();
nodeQueue.add(root);
int index = 1;
while(!nodeQueue.isEmpty()) {
TreeNode node = nodeQueue.remove();
if (index == parts.length) {
break;
}
item = parts[index++];
item = item.trim();
if (!item.equals("null")) {
int leftNumber = Integer.parseInt(item);
node.left = new TreeNode(leftNumber);
nodeQueue.add(node.left);
}
if (index == parts.length) {
break;
}
item = parts[index++];
item = item.trim();
if (!item.equals("null")) {
int rightNumber = Integer.parseInt(item);
node.right = new TreeNode(rightNumber);
nodeQueue.add(node.right);
}
}
return root;
}
public static String integerArrayListToString(List<Integer> nums, int length) {
if (length == 0) {
return "[]";
}
String result = "";
for(int index = 0; index < length; index++) {
Integer number = nums.get(index);
result += Integer.toString(number) + ", ";
}
return "[" + result.substring(0, result.length() - 2) + "]";
}
public static String integerArrayListToString(List<Integer> nums) {
return integerArrayListToString(nums, nums.size());
}
}
上面提供了两种数据类型转换为TreeNode的方法,一种是String类型的数据输入转换为TreeNode,一种是把Integer列表转换String再转换为TreeNode。
归根结底其实就一种方法,之所以最后都用String来转换为TreeNode,是因为在leetcode的题目里都是用null来表示当前节点是空,而int[]是无法储存null值的。
stringToTreeNode这个方法代码还是挺清楚的,唯一关键点是定义了一个 节点的 Queue,来保证是先左后右地插入每个节点数据。