二叉树最长连续序列和树形dp原理

文章资讯 2020-08-07 21:39:47

二叉树最长连续序列和树形dp原理

用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;
}
};