LintCode 614. 二叉树的最长连续子序列 II

614. 二叉树的最长连续子序列 II

给定一棵二叉树，找到最长连续序列路径的长度。
路径起点跟终点可以为二叉树的任意节点。

样例
1
/ \
2 0
/
3
返回 4 // 0-1-2-3

思路：需要注意最后得到的结果必然由一半从下往上增加，另一半从下往上减小的两部分组成。所以对于每个节点都需要计算这两个值。因为需要返回两个值，所以自定义了一个类ReturnInfo，其中的longest_up表示以当前根节点为末节点从下往上增长的最长连续数量，longest_down类似，关键是要理解连续，以及在每个节点的处理细节.

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */

class ReturnInfo{
public:
    int longest_up;
    int longest_down;
    ReturnInfo(int up,int down):longest_up(up),longest_down(down){};
}


class Solution {
public:
    /**
     * @param root: the root of binary tree
     * @return: the length of the longest consecutive sequence path
     */
    int longestConsecutive2(TreeNode * root) {
        // write your code here
        longestInfo(root);
        return longest;
    }

    ReturnInfo longestInfo(TreeNode * root){
        if(root == nullptr) return ReturnInfo(0,0);//nullptr返回(0,0)
        int up = 1;//默认只有根节点，初始化为1
        int down = 1;
        ReturnInfo tempL = longestInfo(root->left);//左右节点的连续信息
        ReturnInfo tempR = longestInfo(root->right);

        if(root->left){
            if(root->left->val == root->val - 1) up = max(up,tempL.longest_up+1);//左子树是否连续进行更新
            if(root->left->val == root->val + 1) down = max(down,tempL.longest_down+1);
        }
        
        if(root->right){
            if(root->right->val == root->val - 1) up = max(up,tempR.longest_up+1);//右子树处理
            if(root->right->val == root->val + 1) down = max(down,tempR.longest_down+1);
        }
        
        longest = max(longest,up+down-1);//以当前root为最高点的连续数字最多数量，仔细想想就是up+down-1。
        return ReturnInfo(up,down);
    }
    
private:
    int longest = 0;
};