1. 程式人生 > >刷題——最長連續序列

刷題——最長連續序列

import java.util.Arrays;
import java.util.HashSet;

/*
 * 題目描述

Given an unsorted array of integers, find the length of the longest consecutive elements sequence.
For example,
Given[100, 4, 200, 1, 3, 2],
The longest consecutive elements sequence is[1, 2, 3, 4]. Return its length:4.
Your algorithm should run in O(n) complexity.
 */
public class 最長連續序列 {
	//方法一:複雜度o(nlogn)	
	/*public int longestConsecutive(int[] num) {
		if (num==null || num.length==0) {
			return 0;
		}
        Arrays.sort(num);
        int maxLength = 1;
        int count = 1;
        for (int i = 1; i < num.length; i++) {
			if (num[i]==num[i-1]+1) {
				count++;
			}else if (num[i]==num[i-1]) {
				continue;
			}else{				
				count = 1;
			}
			maxLength = count>maxLength?count:maxLength;
		}
        return maxLength;
    }*/
	//方法二:使用HashSet,複雜度主要在於遍歷陣列中元素
	public int longestConsecutive(int[] num) {
        HashSet<Integer> set = new HashSet<Integer>();
        //加入元素,同時可以去重,o(n)
        for (int i = 0; i < num.length; i++) {
			set.add(num[i]);
		}
        //從每一個元素往兩邊走,連續的都去除掉,這樣訪問完一遍後所有元素就都刪光了,複雜度o(n)
        int maxLength = 1;
        for (int i = 0; i < num.length; i++) {
        	int count = 1;
			if (!set.isEmpty() && set.remove(num[i])) {//如可以移除此元素,證明還存在於set中
				int down = num[i]-1;
				int up = num[i]+1;
				while (set.remove(down)) {
					count++;
					down--;
				}
				while (set.remove(up)) {
					count++;		
					up++;
				}
				maxLength = count>maxLength?count:maxLength;
			}
		}
        return maxLength;
    }
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		//int[] num = {1,0,1,2};
		int[] num = {100, 4, 200, 1, 3, 2};
		System.out.println(new 最長連續序列().longestConsecutive(num));
	}

}