1. 程式人生 > >leetcode(Two Sum)

leetcode(Two Sum)

Title:Two Sum     1

Difficulty:Easy

原題leetcode地址:https://leetcode.com/problems/two-sum/

 

下面主要是3種解法:

1. 暴力法,時間&空間複雜度如下:

時間複雜度:O(n^2),兩層for迴圈,每一層for迴圈再沒有找到對應的target時,都是要執行n次。

空間複雜度:O(1),申請了一維長度為2陣列。

    /**
     * 暴力法
     * @param nums
     * @param target
     * @return
     */
    public static int[] twoSum(int[] nums, int target) {
        int index[] = new int[]{0, 1};

        for (int i = 0; i < nums.length; i++) {
            for (int j = i + 1; i < nums.length; j++) {
                if (target == nums[i] + nums[j]) {
                    index[0] = i;
                    index[1] = j;
                }
            }
        }

        return index;
    }

2. 採用Set集合的唯一性,時間&空間複雜度如下:

時間複雜度:O(n),雖然程式碼中包含了兩層for迴圈,但是不是在第一層時,第二層每次都需要執行一次for迴圈。

空間複雜度:O(n),申請了HashSet,最長為n。

    /**
     * 利用Set集合的唯一性
     * @param nums
     * @param target
     * @return
     */
    public static int[] twoSum2(int[] nums, int target) {
        int index[] = new int[]{0, 1};
        Set hset = new HashSet();

        for (int i = 0 ; i < nums.length; i++) {
            if (hset.add(target - nums[i])) {
                hset.remove(target - nums[i]);
                hset.add(nums[i]);
            } else {
                index[1] = i;
                for (int j = 0; j < i; j++) {
                    if (target == (nums[i] + nums[j])) {
                        index[0] = j;
                    }
                }
            }
        }

        return index;
    }

3. 採用Map的Key的唯一性,時間&空間複雜度如下:

時間複雜度:O(n),一層for迴圈,最長遍歷時資料的長度。

空間複雜度:O(n),申請了HashMap,最長為n。

    /**
     * 利用Map的Key唯一性
     * @param nums
     * @param target
     * @return
     */
    public static int[] twoSum3(int[] nums, int target) {
        int index[] = new int[]{0, 1};
        Map<Integer, Integer> hmap = new HashMap<Integer, Integer>();

        for (int i = 0; i < nums.length; i++) {
            if (hmap.containsKey(target - nums[i])) {
                index[0] = hmap.get(target - nums[i]);
                index[1] = i;
            }
            hmap.put(nums[i], i);
        }

        return index;
    }