当前位置: 代码迷 >> 综合 >> 第三章 二叉树问题(在二叉树中找到累加和为指定值的最长路路径长度)
  详细解决方案

第三章 二叉树问题(在二叉树中找到累加和为指定值的最长路路径长度)

热度:70   发布时间:2023-10-11 12:13:19.0

【题目】

        给定一棵二叉树的头节点head和一个32位整数sum,二叉树节点值类型为整型,求累加和为sum的最长路径长度。路径是指从某个节点往下,每次最多选择一个孩子节点或者不选所组成的节点链。

package base;import java.util.HashMap;
import java.util.Map;public class P119_BtMaxLen {/*** 获取最长路径长度* @param head 头节点* @param k 指定值* @return*/public static int getMaxLen(Node head, int k) {Map<Integer, Integer> sumMap = new HashMap<Integer, Integer>();sumMap.put(0, 0);//重要return preOrder(head, k, 0, 1, 0, sumMap);}/*** 左子数和右子数 分别提供 最长路径* @param node* @param k* @param preSum 父节点累加和* @param level  节点层数* @param maxLen 最长路径* @param sumMap* @return*/public static int preOrder(Node node, int k, int preSum, int level, int maxLen, Map<Integer, Integer> sumMap) {if (node == null) {return maxLen;}int curSum = preSum + node.value;if (!sumMap.containsKey(curSum)) {sumMap.put(curSum, level);}if (sumMap.containsKey(curSum - k)) {maxLen = Math.max(level - sumMap.get(curSum - k), maxLen);}int leftMaxLen = preOrder(node.left, k, curSum, level + 1, maxLen, sumMap);int rightMaxLen = preOrder(node.right, k, curSum, level + 1, maxLen, sumMap);if (level == sumMap.get(curSum)) {sumMap.remove(curSum);// 当前节点返回上一层时,数据已无用,需要删除}return Math.max(leftMaxLen, rightMaxLen);}public static void main(String[] args) {Node node = new Node(-3);node.left = new Node(3);node.left.left = new Node(1);node.left.right = new Node(0);node.left.right.left = new Node(1);node.left.right.right = new Node(6);node.right = new Node(9);node.right.left = new Node(2);node.right.right = new Node(1);int maxLen = getMaxLen(node, 6);System.out.println(maxLen);}public static class Node {public int value;public Node left;public Node right;public Node(int value) {this.value = value;}}
}

        

  相关解决方案