• OptimalSolution(2)--二叉树问题(3)Path路径问题


      一、在二叉树中找到累加和为指定值的最长路径长度

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

                  -3
                /    
               3      -9
              /      / 
             1    0  2   1  
                  /               如果sum=6,那么累加和为6的最长路径为:-3,3,0,6,所以返回4
                 1  6              如果sum=-9,那么累加和为-9的最长路径为:-9,所以返回1

      第一步:生成变量maxLen,记录累加和等于sum的最长路径长度

      第二步:生成哈希表sumMap,负责记录从根结点root开始的一条路径上的累加和出现情况。sumMap的key值代表某个累加和,value值代表这个累加和在路径中最早出现的层数。如果在遍历到cur节点的时候,已经知道了从root到cur节点这条路径上的累加和出现情况,那么求以cur节点结尾的累加和为指定值的最长路径长度就非常容易。那么问题是,如何去更新sumMap,才能够做到在遍历到任何一个节点的时候都能有从root到这个节点的路径上的累加和出现情况?

      第三步:首先在sumMap中加入一个记录(0,0),表示累加和0不用包括任何节点就可以得到。然后先序遍历节点,遍历到的当前节点记为cur,从root到cur父节点的累加和记为preSum,cur所在的层数记为level,将cur.val+preSum值记为curSum,就是从root到cur的累加和。如果sumMap中已经包含了curSum的记录,说明curSum在上层已经出现过,那么就不更新sumMap;如果sumMap不包含curSum的记录,说明curSum是第一次出现,就把(curSum,level)这个记录放入sumMap。在以cur为头节点的子树处理完,在返回cur父节点之前,还需要在sumMap中查询curSum这个累加和(key)出现的层数(value),如果value等于level,说明curSum这个累加和的记录是在遍历到cur时加上去的,那么就把这条记录删除,如果value不等于level,则不做任何调整。

       分析执行过程:

    0.{0=0},curSum=0,
    1.-3,{0=0, -3=1},curSum=-3,
    2.-3 → 3,{0=0, -3=1},curSum=0,
    3.-3 → 3 → 1,{0=0, 1=3, -3=1},curSum=1,
    4.-3 → 3,{0=0, -3=1},curSum=0,
    5.-3 → 3 → 0,{0=0, -3=1},curSum=0,
    6.-3 → 3 → 0 → 1,{0=0, 1=4, -3=1},curSum=1,
    7.-3 → 3 → 0, {0=0, -3=1},curSum=0,
    8.-3 → 3 → 0 → 6, {0=0, -3=1, 6=4},curSum=6,
    9.-3 → 3 → 0,{0=0, -3=1},curSum=0,
    10.-3 → 3,{0=0, -3=1},curSum=0,
    11.-3 ,{0=0, -3=1},curSum=0,
    12.-3 → -9,{0=0, -3=1, -12=2},curSum=-12,
    13.-3 → -9 → 2,{0=0, -3=1, -10=3, -12=2},curSum=-10,
    14.-3 → -9 ,{0=0, -3=1, -12=2},curSum=-12,
    15.-3 → -9 → 1,{0=0, -3=1, -11=3, -12=2},curSum=-11,
    16.-3 → -9 ,{0=0, -3=1, -12=2},curSum=-12,
    17.-3 ,{0=0, -3=1},curSum=-3,
    18.{0=0,curSum=0,

      情况1:sum=6,根据 if ( sumMap.containsKey(curSum - sum)) { maxLen = Math.max(level - sumMap.get(curSum - sum), maxLen); }在上面的第8步,curSum-sum=0,maxLen=0,level-sumMap.get(curSum - sum)=4-0=0,因此maxLen=4。

      情况2:sum=-8,在上面的第15步,curSum=-11,而curSum-sum=-3,level-sumMap.get(curSum - sum)=3-1=2,因此maxLen=2,这里的意思就是,从根节点出发累加和为-11的时候需要走3level,而根据sumMap中的-3=1可以知道,从根节点出发累加和为-3的时候只需要走1level,因此要想实现累加和为-8只需要从1level走到3level即可。

      代码实现:

        public int getMaxLength(Node root, int sum) {
            // key表示某个累加和,value表示这个累加和在路径中最早出现的层数
            HashMap<Integer, Integer> sumMap = new HashMap<>();
            // 表示累加和0不用经过任何节点就可以得到
            sumMap.put(0, 0);
            return preOrder(root, sum, 0, 1, 0, sumMap);
        }
        
        /**
         * @param root        遍历到当前节点,记为cur
         * @param sum        要求的累加和
         * @param preSum    从整棵树的根节点root到cur的父节点累加和
         * @param level        cur所在的层数
         * @param sumMap    记录累加和与该累加和在路径中最早出现的层数
         * @return maxLen    累加和等于sum的最长路径长度
         */
        private int preOrder(Node root, int sum, int preSum, int level, int maxLen, HashMap<Integer, Integer> sumMap) {
            if (root == null) return maxLen;
            // 从root到cur的累加和等于从root到cur的父节点的累加和加上cur的val
            int curSum = preSum + root.val;
            // 如果sumMap不包含这个累加和,说明是第一次出现,就加入到sumMap中
            if (!sumMap.containsKey(curSum)) sumMap.put(curSum, level);
            // 如果curSum - sum已经包含了,就说明之前已经有了从根节点到达curSum - sum需要的level数,因此,
            // 用从根节点到当前节点的level数减去从根节点到curSum - sum需要的level数就是从curSum - sum节点走到当前curSum节点的level数
            // 也就是,curSum - (curSum - sum) = sum
            if ( sumMap.containsKey(curSum - sum)) {
                maxLen = Math.max(level - sumMap.get(curSum - sum), maxLen);
            }
            maxLen = preOrder(root.left,  sum, curSum, level + 1, maxLen, sumMap);
            maxLen = preOrder(root.right, sum, curSum, level + 1, maxLen, sumMap);
            // 如果curSum这个累加和的记录是在遍历到cur时加上去的,那么在返回cur的父节点之前需要删除这个记录,
            // 否则就不是从cur的父节点到cur的父节点的其他子节点,而是从cur到cur的兄弟节点
            if (level == sumMap.get(curSum)) sumMap.remove(curSum);
            return maxLen;
        }

      二、在二叉树中找到两个节点的最近公共祖先

      1.给定一棵二叉树以及这棵树的两个节点o1和o2,返回o1和o2的最近公共祖先节点。

                   1
                /    
               2       3
              /      / 
             4    5  6   7  
                         /
    8 4和5的最近公共祖先为2,5和2的最近公共祖先为2,6和8的最近公共祖先为3,5和8的最近公共祖先为1

      解法:后序遍历二叉树,假设遍历到的当前节点为cur,因为后序遍历要先处理cur的两棵子树,假设处理cur左子树时返回节点为left,处理右子树时返回节点为right

      (1)如果cur等于null或者o1或者o2,返回cur

      (2)如果left和right都为空,说明cur整棵子树上都没有发现过o1和o2,返回null

      (3)如果left和right都不为空,说明左子树上发现过o1或o2,右子树上也发现过o2或o1,说明o1向上与o2向上的过程中,首次在cur相遇,返回cur

      (4)如果left和right有一个为空,有一个不为空,假设不为空的节点记为node,此时有两种可能,要么node是o1或o2中的一个,要么node已经是o1和o2的最近公共祖先,两种情况下,都可以返回node。

      执行过程是:

    假设o1=6,o2=8
    依次遍历4,5,2都没有发现o1或o2,所以1的左子树返回null
    遍历6,发现等于o1,返回6,所以3的左子树返回6
    遍历8,发现等于o2,返回8,所以7的左子树返回8
    结点7的右子树为null,而左子树为8,所以返回8
    遍历3,左子树返回6,右子树返回8,此时返回3
    遍历1,左子树返回null,右子树返回3,因此返回3

      代码实现:

        public Node lowestAncestor(Node root, Node o1, Node o2) {
            // 情况1:root为null时返回null,情况2:root为o1或者o2本身时,返回root
            if (root == null || root == o1 || root == o2) return root;
            Node left  = lowestAncestor(root.left, o1, o2);
            Node right = lowestAncestor(root.right, o1, o2);
            // 情况3,在左子树发现了o1或o2,在右子树发现了o1或o2,那么最近公共祖先一定是root,返回root
            if (left != null && right != null) return root;
            // 情况4,left和right如果都为null,那么返回null
            // 情况5:left和right有且只有一个不为null,那么返回不为null的那个
            return left != null ? left : right;
        }

      2.

      3.  

      三、求二叉树节点间的最大距离

      问题:从二叉树的节点A出发,沿途的节点只能经过一次,当到达节点B时,路径上的节点数叫做A到B的距离,包括A和B本身。

      思路:一个以root为根节点的树,最大距离只能来自以下三种情况:(1)root的左子树上的最大距离(2)root的右子树上的最大距离(3)root左子树上例root.left最远的距离 + 1 + root的右子树上离root.right最远的距离

      解法:1.整个过程为后序遍历。2.假设子树根节点为root,处理root左子树,得到两个信息,左子树上的最大距离记为lMax,左子树上距离root左孩子的最远距离记为maxfromLeft。同理,右子树上最大距离为rMax,右子树上距离root右孩子最远距离记为maxFromRight。那么maxFromLeft + 1 + maxFromRight就是跨root节点的最大距离,再与lMax和rMax比较,把三者中最大值作为root树上最大距离返回,maxFromLeft+1就是root左子树上离root最远的点到root的距离,maxFromRight+1就是root右子树上离root最远的点到root的距离,选两者中最大的一个作为root树上距离root最远的距离返回。

      执行过程是:

                   1
                /    
               2       3
              /       /4       6     
    遍历4,lMax=0,record[0]=0,rMax=0,record[0]=0,curNodeMax=1,record[0]=1,返回1
    遍历2,lMax=1,record[0]=1,rMax=0,record[0]=0,curNodeMax=2,record[0]=2,返回2
    遍历6,lMax=0,record[0]=0,rMax=0,record[0]=0,curNodeMax=1,record[0]=1,返回1
    遍历3,lMax=1,record[0]=1,rMax=0,record[0]=0,curNodeMax=2,record[0]=2,返回2
    遍历1,lMax=2,record[0]=2,rMax=2,record[0]=2,curNOdeMax=5,record[0]=3,返回5,结束

      代码实现:

        public int maxDistance(Node root) {
            int[] record = new int[1];
            return postOrder(root, record);
        }
        
        public int postOrder(Node root, int[] record) {
            if (root == null) {
                record[0] = 0;
                return 0;
            }
            int lMax = postOrder(root.left , record);                  // 左子树上两点最远距离
            int maxFromLeft  = record[0];                              // 到root.left最远距离
            int rMax = postOrder(root.right, record);                  // 右子树上两点最远距离
            int maxFromRight = record[0];                              // 到root.right最远距离
            int curNodeMax = maxFromLeft + 1 + maxFromRight;           // 经过root的最远距离
            record[0] = Math.max(maxFromLeft, maxFromRight) + 1;       // 到root的最远距离
            return Math.max(Math.max(lMax, rMax), curNodeMax);          // 返回以root为根节点的树的两节点的最大距离
        }

      四、求根节点到叶节点的路径

                  -3
                /    
               3      -9
              /      / 
             1    0  2   1  
                  /              1  6              输出:[-3->3->1, -3->3->0->1, -3->3->0->6, -3->-9->2, -3->-9->1]

      分析过程:前序遍历,使用两个全局变量List<String> res, String str,其中res作为结果集list,str作为字符串。

    1.""开始
    2.-3->,遍历到-3
    3.-3->3,遍历到3
    4.-3->3->1->,遍历到1,打印 5.-3->3->0->1->遍历到1,打印 6.-3->3->0->6->遍历到6,打印 7.-3->3->0->遍历回到0 8.-3->3->遍历回到3 9.-3->-9->2->遍历到2,打印 10.-3->-9->1->遍历到1,打印 11.-3->-9->遍历回到-9 12.-3->遍历回到-3
    13."" 结束

      代码实现:

        public List<String> binaryTreePaths(TreeNode root) {
            List<String> res = new ArrayList<>();
            preOrder(root, res, "");
            return res;
        }
        
        private void preOrder(TreeNode root, List<String> res, String str) {
            if (root == null) return ;
            str += root.val + "->";
            preOrder(root.left , res, str);
            preOrder(root.right, res, str);
            if (root.left == null && root.right == null) {
                str = str.substring(0, str.length() - 2);
                res.add(str);
            }
        }

       变体:求是否存在从根节点到叶节点的节点和等于给定的值sum的路径

        public boolean hasPathSum(TreeNode root, int sum) {
            boolean[] res = new boolean[1];
            preOrder(root, sum, 0, res);
            return res[0];
        }
        
        private void preOrder(TreeNode root, int sum, int tmp, boolean[] res) {
            if (root == null) return;
            tmp += root.val;
            preOrder(root.left,  sum, tmp, res);
            preOrder(root.right, sum, tmp, res);
            if (root.left == null && root.right == null && tmp == sum) {
                res[0] = true;
                return ;
            }
        }

       变体:求一棵树上从上到下的路径数,这个路径满足节点的累加和等于给定值sum(需要注意pathSum还要自循环)

        public int pathSum(TreeNode root, int sum) {
            if (root == null) return 0;
            int[] count = new int[1];
            preOrder(root, sum, 0, count);
            return count[0] + pathSum(root.left, sum) + pathSum(root.right, sum);
        }
        
        private void preOrder(TreeNode root, int sum, int tmp, int[] count) {
            if (root == null) return ;
            tmp += root.val;
            preOrder(root.left , sum, tmp, count);
            preOrder(root.right, sum, tmp, count);
            if (tmp == sum) {
                count[0]++;
            }
        }
  • 相关阅读:
    rabbitMq(2)之“Work”模式
    rabbitmq(1)之"Hello World"模式
    Maven的实用基础命令
    RabbitMq(3)之“发布/订阅”模式
    mac rabbitmq安装以及入门
    Mac 安装Homebrew
    多线程通讯之管道
    CentOS中防火墙之iptables
    CentOS中防火墙之Firewall
    Redis的简介
  • 原文地址:https://www.cnblogs.com/BigJunOba/p/9630525.html
Copyright © 2020-2023  润新知