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

【题目】

        给定一棵二叉树的头节点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;
        }
    }
}