Tony's Log

Algorithms, Distributed System, Machine Learning

  博客园 :: 首页 :: 博问 :: 闪存 :: 新随笔 :: 联系 :: 订阅 订阅 :: 管理 ::

A simple application to binary tree traversal.

class Solution {
    typedef pair<int, unsigned> Rec; // last val - length
    unsigned ret;
    void go(TreeNode *p, Rec r)
    {
        if(!p) return;
        
        Rec nr(p->val, 1);
        if(r.second && p->val == (r.first + 1))
        {
            nr = Rec(p->val, r.second + 1);
        }
        ret = max(ret, nr.second);
        
        go(p->left, nr);
        go(p->right, nr);
    }
public:
    int longestConsecutive(TreeNode* root) {
        ret = 0;
        go(root, Rec(0, 0));
        return ret;
    }
};
posted on 2015-10-28 11:47  Tonix  阅读(160)  评论(0编辑  收藏  举报