298. Binary Tree Longest Consecutive Sequence

最后更新

三刷。
11-Jan-2017

简单的二叉树，没啥技巧可言，DFS..

Time: O(n)
Space: O(depth)

public class Solution {
    int max = 0;
    public int longestConsecutive(TreeNode root) {
        if (root == null) return max;
        
        dfs(root, 0);
        return max;
    }
    
    public void dfs(TreeNode root, int depth) {
        max = Math.max(max, ++ depth);
        if (root == null) return;
        if (root.left == null && root.right == null) return;
        
        if (root.left != null) {
            if (root.val + 1 == root.left.val) {
                dfs(root.left, depth);
            } else {
                dfs(root.left, 0);
            }
        }
        
        if (root.right != null) {
            if (root.val + 1 == root.right.val) {
                dfs(root.right, depth);
            } else {
                dfs(root.right, 0);
            }
        }
    }
}

---

好像就是楞做，记录长度就行了。
判断失败就从0开始。。

public class Solution 
{
    int max = 0;
    public int longestConsecutive(TreeNode root) 
    {
        if(root == null) return 0;

        helper(root,0);
        return max;
    }
    
    public void helper(TreeNode root, int c)
    {
        max = Math.max(max,c+1);
        if(root.left == null && root.right == null) return;
        if(root.left != null)
        {
            if(root.val+1 == root.left.val) helper(root.left,c+1);
            else helper(root.left,0);
            
        }
        
        if(root.right != null)
        {
            if(root.val+1 == root.right.val) helper(root.right,c+1);
            else helper(root.right,0);
            
        }
    }
}

iterative比较难啊。。貌似得记录每一个NODE的当时情况，就得有个MAP之类的了。。

先不考虑了 二刷 再说……

---

---

二刷。

DFS，不连续了就更新一下。

二刷做的不如一刷好，写得不如一刷简洁。

public class Solution {
    int res;
    public int longestConsecutive(TreeNode root) {
        if (root == null) return 0;
        this.res = 1;
        twoShots(root.left, root.val, 1);
        twoShots(root.right, root.val, 1);
        return res;
    }
    
    public void twoShots(TreeNode root, int prev, int num) {
        res = Math.max(res, num);
        if (root == null) return;
        if (root.val == prev + 1) {
            twoShots(root.left, prev + 1, num + 1);
            twoShots(root.right, prev + 1, num + 1);
        } else {
            twoShots(root.left, root.val, 1);
            twoShots(root.right, root.val, 1);
        }
    }
}