二叉树中的最长交错路径

二叉树中的最长交错路径


给你一棵以 root 为根的二叉树,二叉树中的交错路径定义如下:

选择二叉树中 任意 节点和一个方向(左或者右)。 如果前进方向为右,那么移动到当前节点的的右子节点,否则移动到它的左子节点。
改变前进方向:左变右或者右变左。 重复第二步和第三步,直到你在树中无法继续移动。 交错路径的长度定义为:访问过的节点数目 -
1(单个节点的路径长度为 0)。

请你返回给定树中最长 交错路径 的长度。

题解
动态规划
每遍历到一个节点就需要当前点的最长交错路径
定义dp[2]
dp[0]表示当前节点下一步走左子树的交错路径数
dp[1]表示当前节点下一步走右子树的交错路径数

dp[0]的交错路径数就是其右子树的交错路径数+1
dp[1]的交错路径数就是其左子树的交错路径数+1

使用变量记录下最大的交错路径数。
自此形成一个自底向上的dfs

public class longestZigZag {
    private int maxPath=0;
    public int longestZigZag(TreeNode root){
        dfs(root);
        return maxPath;
    }

    private int[] dfs(TreeNode root){
        int[] dp=new int[2];
        if (root==null){
            dp[0]=-1;
            dp[1]=-1;
            return dp;
        }
        int[] left=dfs(root.left);
        int[] right=dfs(root.right);
        dp[0]=1+left[1];
        dp[1]=1+right[0];
        maxPath=Math.max(maxPath,Math.max(dp[0],dp[1]));
        return dp;
    }
}

猜你喜欢

转载自blog.csdn.net/qq_30926503/article/details/107611592