參考網址:https://leetcode.com/problems/longest-univalue-path/discuss/108136/JavaC%2B%2B-Clean-Code
解題思路:
(1)想到二叉樹,可能最常用的就是遞歸,以下主要解釋代碼的兩個部分
(2)return max(resl, resr);表示的是一條路徑,因此不能有分支。
(3)所以我們對於一個結點而言,選擇的最長路徑是在左右子樹中選擇的
(4)但是如果這分支是在根結點,那麼就可以接受,lup = max(lup, resl + resr)就是做這件事的
(5)存在即合理,如果存在某個節點,以它爲轉折點,長度大於目前的lup,那麼更新lup的值
(6)但是不排除後續會有更大的值,這裏lup只是表示存在某條路徑,我們並不記錄
(7)這一技巧和前面的Longest Increasing Subsequence(C++最長上升子序列)很像,網址如下:
(8)https://blog.csdn.net/coolsunxu/article/details/105403508
class Solution {
public:
int longestUnivaluePath(TreeNode* root) {
int lup = 0;
if (root) dfs(root, lup);
return lup;
}
private:
int dfs(TreeNode* node, int& lup) {
int l = node->left ? dfs(node->left, lup) : 0;
int r = node->right ? dfs(node->right, lup) : 0;
int resl = node->left && node->left->val == node->val ? l + 1 : 0;
int resr = node->right && node->right->val == node->val ? r + 1 : 0;
lup = max(lup, resl + resr);
return max(resl, resr);
}
};