力扣labuladong——一刷day46

本文介绍了力扣网站上的三个二叉树相关题目:971题翻转二叉树匹配先序遍历,987题二叉树的垂序遍历,以及666题路径总和IV。通过递归方法详细解析了每个问题的解决方案。

提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档


前言


二叉树的递归分为「遍历」和「分解问题」两种思维模式,这道题需要用到「遍历」的思维。

一、力扣971. 翻转二叉树以匹配先序遍历

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    List<Integer> res = new ArrayList<>();
    int[] voyageArr;
    int i = 0;
    boolean flag = true;
    public List<Integer> flipMatchVoyage(TreeNode root, int[] voyage) {
        voyageArr = voyage;
        fun(root);
        if(flag == false){
            List<Integer> list = new ArrayList<>();
            list.add(-1);
            return list;
        }
        return res;
    }
    public void fun(TreeNode root){
        if(root == null){
            return;
        }
        if(root.val != voyageArr[i++]){
            flag = false;
            return;
        }
        if(root.left != null && root.left.val != voyageArr[i]){
            res.add(root.val);
            TreeNode temp = root.left;
            root.left = root.right;
            root.right = temp;
        }
        fun(root.left);
        fun(root.right);
    }
}

二、力扣987. 二叉树的垂序遍历

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    class Triple{
        TreeNode node;
        int row, col;
        public Triple(TreeNode node, int row, int col){
            this.node = node;
            this.row = row;
            this.col = col;
        }
    }
    LinkedList<List<Integer>> res = new LinkedList<>();
    List<Triple> nodes = new ArrayList<>();
    public List<List<Integer>> verticalTraversal(TreeNode root) {
        traverse(root,0,0);
        Collections.sort(nodes, (tri1,tri2)->{
            if(tri1.col != tri2.col){
                return tri1.col - tri2.col;
            }else if(tri1.row != tri2.row){
                return tri1.row - tri2.row;
            }else{
                return tri1.node.val - tri2.node.val;
            }
        });
        int pre = Integer.MIN_VALUE;
        for(int i = 0; i < nodes.size(); i ++){
            Triple cur = nodes.get(i);
            if(cur.col != pre){
                res.addLast(new LinkedList<Integer>());
                pre = cur.col;
            }
            res.getLast().add(cur.node.val);
        }
        return res;
    }
    public void traverse(TreeNode root, int row, int col){
        if(root == null){
            return ;
        }
        nodes.add(new Triple(root,row,col));
        traverse(root.left, row+1, col-1);
        traverse(root.right, row+1, col + 1);
    }
}

三、力扣666. 路径总和 IV

class Solution {
    Map<Integer,Integer> tree = new HashMap<>();
    int sum = 0;
    public int pathSum(int[] nums) {
        for(int i = 0; i < nums.length; i ++){
            int value = nums[i]%10;
            int code = nums[i]/10;
            tree.put(code,value);
        }
        int rootCode = nums[0]/10;
        fun(rootCode,0);
        return sum;
    }
    public void fun(int code, int path){
        if(!tree.containsKey(code)){
            return;
        }
        int value = tree.get(code);
        int[] pos = decode(code);
        int depth = pos[0], id = pos[1];
        int leftCode = encode(depth+1, 2*id-1);
        int rightCode = encode(depth+1,2*id);
        if(!tree.containsKey(leftCode) && !tree.containsKey(rightCode)){
            sum += path + value;
            return;
        }
        fun(leftCode, path + value);
        fun(rightCode, path + value);
    }
    public int[] decode(int code){
        int id = code%10;
        int depth = code/10;
        return new int[]{depth,id};
    }
    public int encode(int depth, int id){
        return depth*10 + id;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

乱世在摸鱼

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值