😮128 longest consecutive sequence

https://leetcode.com/problems/longest-consecutive-sequence/

class Solution {
    public int longestConsecutive(int[] nums) {
        Set<Integer> numSet = new HashSet<>();
        for (int num : nums) {
            numSet.add(num);
        }
        int longestStreak = 0;
        for (int num: nums) {
            // 只考虑能作为起始的数
            if (!numSet.contains(num - 1)) {
                int curNum = num;
                int streak = 1;
                while (numSet.contains(curNum + 1)) {
                    streak++;
                    curNum++;
                }
                longestStreak = Math.max(longestStreak, streak);
            }
        }
        return longestStreak;
    }
}

找到每个可能是起点的数(num-1不在list里), 然后往后数到下一个数不在list里,记录长度,amortize下来时间复杂度是O(n)的

所以当条件不确定,比如一个数可以是升序起始或者降序起始时,要考虑固定一端的条件。

也可以用union find

class Solution {
    class Union {
        int[] size;
        int[] leader;
        Union(int[] nums) {
            size = new int[nums.length];
            leader = new int[nums.length];
            for (int i = 0; i < nums.length; i++) {
                leader[i] = i;
                size[i] = 1;
            }
        }
        int union(int a, int b) {
            int la = find(a);
            int lb = find(b);
            if (la == lb) {
                return size[la];
            }
            if (size[la] >= size[lb]) {
                size[la] += size[lb];
                leader[lb] = la;
                return size[la];
            } else {
                size[lb] += size[la];
                leader[la] = lb;
                return size[lb];
            }
        }
        int find(int a) {
            if (leader[a] == a) {
                return a;
            }
            leader[a] = find(leader[a]);
            return leader[a];
        }
    }
    public int longestConsecutive(int[] nums) {
        Union union = new Union(nums);
        Map<Integer, Integer> map = new HashMap();
        int len = 0;
        for (int i = 0; i < nums.length; i++) {
            map.put(nums[i], i);
        }
        for (int num : nums) {
            if (map.containsKey(num-1)) {
                len = Math.max(len, union.union(map.get(num), map.get(num-1)));
            }
            if (map.containsKey(num+1)) {
                len = Math.max(len, union.union(map.get(num), map.get(num+1)));
            }
            len = Math.max(len, union.size[map.get(num)]);
        }
        return len;
    }
}

Last updated