LeetCode 687 -- 二叉树

题目描述

最长同值路径
题目要求我们求树中两个节点之间的路径长度，并且满足这两个节点之间的所有节点的值相等，这个长度由它们之间的边数表示，其实就是节点数减一。
注意，这两个节点是可以跨越根节点的，也就是说，这两个节点不必在同一个子树中！

思路

对于二叉树的题目，🌲的遍历我们肯定是逃不过取的，问题是，二叉树有四种遍历方式，我们应该采用那种呢？
对于本题而言，后续遍历是一种合理的方式，因为我们需要在根节点汇总信息，将跟节点的值与左子树和右子树的值去比较，如果相等的话，说明跟节点和子树可以连接在一起。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int res = 0;
    
    int dfs(TreeNode *root)
    {
        if(root == NULL)   return 0; // 好习惯
        
        int left = dfs(root->left), right = dfs(root->right); // 这里求的是以左/右孩子为跟节点的最大路经常
        int retl = 0, retr = 0; // 这里求的是以当前节点为跟节点并且只与一个子树连通的路经长
        if(root->left && root->left->val == root->val)  retl = left + 1;
        if(root->right && root->right->val == root->val)    retr = right + 1;
        
        res = max(res, retl + retr); // 总的路经长是左子树+右zishu
        return max(retl, retr);  // 返回的是左子树 or 右子树的最大路径长度，倒y结构是不合法的，return (retl + retr) 是错误的！
    }
    
    int longestUnivaluePath(TreeNode* root) {
        dfs(root);
        return res;
    }
};