树的学习

树的学习缘由

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,来保证是先左后右地插入每个节点数据。

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值