Longest Univalue Path(C++最長同值路徑)

參考網址: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);
    }
};

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章