1. 程式人生 > 其它 >【LeetCode】1.兩數之和

【LeetCode】1.兩數之和

1.兩數之和

知識點:陣列,雜湊表

題目描述

給定一個整數陣列 nums 和一個整數目標值 target,請你在該陣列中找出 和為目標值 target 的那兩個整數,並返回它們的陣列下標。

你可以假設每種輸入只會對應一個答案。但是,陣列中同一個元素在答案裡不能重複出現。

你可以按任意順序返回答案。

示例
輸入:nums = [2,7,11,15], target = 9
輸出:[0,1]
解釋:因為 nums[0] + nums[1] == 9 ,返回 [0, 1] 。

輸入:nums = [3,2,4], target = 6
輸出:[1,2]

輸入:nums = [3,3], target = 6
輸出:[0,1]

解法一:暴力法

直接進行兩邊陣列的遍歷;

class Solution {
    public int[] twoSum(int[] nums, int target) {
        int[] ans = new int[2];
        for (int i = 0; i < nums.length-1; i++){
            for(int j = i + 1; j < nums.length; j++){
                if (nums[i] + nums[j] == target){
                    ans[0] = i;
                    ans[1] = j;
                    return ans;
                }
            }
        }
        return null;
    }
}

時間複雜度:O(N^2);

解法二:雜湊表

仔細想一下,我們上面的開銷主要在於尋找target-nums[i]這個過程,這個過程需要將整個陣列遍歷;雜湊表的話在尋找值的時候複雜度是O(1);所以可以用雜湊表存起來,然後再在雜湊表裡查詢;

class Solution {
    public int[] twoSum(int[] nums, int target) {
        HashMap<Integer,Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++){
            //檢驗是否存在期望的值;存在直接返回;這個過程的複雜度是O(1);
            //notes:以後刷題不要新建陣列,直接在最後的返回時匿名物件就可以了;
            if (map.containsKey(target - nums[i])) return new int[]{map.get(target - nums[i]), i};
            //不存在加入到雜湊表中;
            else map.put(nums[i], i);
        }
        return null;
    }
}

時間複雜度:O(N);
空間複雜度:O(N);雜湊表的開銷;

體會

其實我們經常會用到雜湊表的containsKey()的這個方法;因為它的檢索時間複雜度為常數;