二叉树最长连续序列 树形dp问题

二叉树最长连续序列 树形dp问题_第1张图片

用f(p)表示从p节点出发的最长连续序列,显然由如下关系

f(p)=max(f(p->left), f(p->right))+1, 

然后再找到f(p)的最大值即可

class Solution {
public:
    int res = 0;
    int longestConsecutive(TreeNode* root) {
        // 这里求的是所有root中最长路径
        dfs(root);
        return res;
    }

    int dfs(TreeNode* root){
        // dfs 定义是从当前root出发的最长连续路径
        if(root==NULL) return 0;
        int left = dfs(root->left);
        int right = dfs(root->right);
        int f_left = (root->left&&root->val+1==root->left->val?left:0);
        int f_right = (root->right&&root->val+1==root->right->val?right:0);
        int count = max(f_left,f_right)+1;
        res = max(res,count);
        return count;
    }
};

 

你可能感兴趣的:(算法)