https://leetcode-cn.com/problems/longest-univalue-path/
// Problem: LeetCode 687
// URL: https://leetcode-cn.com/problems/longest-univalue-path/
// Tags: Tree Recursion DFS
// Difficulty: Easy
#include <iostream>
#include <algorithm>
using namespace std;
struct TreeNode{
int val;
TreeNode* left;
TreeNode* right;
};
class Solution{
private:
int maxLen = 0; // 全局最长同值路径
// 返回该树与根节点同值的路径(必须过根节点)的最大长度
int dfs(TreeNode* root){
// 空树则长度为0
if (root == nullptr)
return 0;
// 该树根节点在左右方向上的最长同值路径的长度(从根节点开始,路径上的节点的值都与根节点一致)
int left = dfs(root->left);
int right = dfs(root->right);
left = (root->left != nullptr && root->val == root->left->val) ? left + 1 : 0;
right = (root->right != nullptr && root->val == root->right->val) ? right + 1 : 0;
// 更新全局最长同值路径
maxLen = max(maxLen, left + right); // 将一条路径分为左右两半,两个结点之间路径长度等于它们到根节点的距离之和
//
return max(left, right);
}
public:
// 求该树中的最长同值路径
int longestUnivaluePath(TreeNode* root) {
dfs(root);
return this->maxLen;
}
};
作者:@臭咸鱼
转载请注明出处:https://www.cnblogs.com/chouxianyu/
欢迎讨论和交流!
原文:https://www.cnblogs.com/chouxianyu/p/13385101.html