687. Longest Univalue Path*

687. Longest Univalue Path*

https://leetcode.com/problems/longest-univalue-path/

题目描述

C++ 实现 1

该题的程序结构有点像 543. Diameter of Binary Tree*

在下面代码中, 我们使用 path_length(root) 这个函数来统计以 root 为起点的 Univalue Path 的最长是多少. 基本思路上, 求左右子树的最大长度, 然后判断左右孩子是否和 root 的值相等, 并更新以 root 为起点的最大长度.

但是, 该函数的返回值不是题目要求的结果, 因为如果 root 的值和左右孩子都相等, 那么 Univalue Path 的长度可以是左右子树的最大长度之和. 所有还会利用一个全局变量, 保存在递归过程中遇到的 longest univalue path.

具体都在代码里, 用文字描述难度较大. 本题解答参考: https://leetcode.com/problems/longest-univalue-path/solution/

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
private:
    int max_length = 0;
    int path_length(TreeNode *root) {
        if (!root) return 0;
        auto a = path_length(root->left);
        auto b = path_length(root->right);
        int path_left = 0, path_right = 0;
        if (root->left && root->left->val == root->val)
            path_left += a + 1;
        if (root->right && root->right->val == root->val)
            path_right += b + 1;
        max_length = std::max(max_length, (path_left + path_right));
        return std::max(path_left, path_right);
    }
public:
    int longestUnivaluePath(TreeNode* root) {
        if (!root) return 0;
        path_length(root);
        return max_length;
    }
};