LeetCode之每日一題(兩數之和)

題目:

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

你可以假設每種輸入只會對應一個答案。但是,你不能重複利用這個數組中同樣的元素。

示例:

給定 nums = [2, 7, 11, 15], target = 9

因爲 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

Java解法:

tip1:

暴力法

暴力法很簡單,遍歷每個元素 x,並查找是否存在一個值與 target - x 相等的目標元素。

public int[] twoSum(int[] nums, int target) {
    for (int i = 0; i < nums.length; i++) {
        for (int j = i + 1; j < nums.length; j++) {
            if (nums[j] == target - nums[i]) {
                return new int[] { i, j };
            }
        }
    }
    throw new IllegalArgumentException("No two sum solution");
}

時間複雜度與空間複雜度:

因爲是雙層循環,所以時間複雜度爲O(n²),而全局分配的空間沒有發生改變,所以空間複雜度爲O(1)。

tip2:

哈希算法

插入的表的同時,檢查是否有符合條件的值,如果有就返回下標。

public int[] twoSum(int[] nums, int target) {
    Map<Integer, Integer> map = new HashMap<>();
    for (int i = 0; i < nums.length; i++) {
        int complement = target - nums[i];
        if (map.containsKey(complement)) {
            return new int[] { map.get(complement), i };
        }
        map.put(nums[i], i);
    }
    throw new IllegalArgumentException("No two sum solution");
}
  • 時間複雜度:O(n)O(n), 我們只遍歷了包含有 nn 個元素的列表一次。在表中進行的每次查找只花費 O(1)O(1) 的時間。

  • 空間複雜度:O(n)O(n), 所需的額外空間取決於哈希表中存儲的元素數量,該表最多需要存儲 nn 個元素。

PS:上面代碼塊是直接執行並返回的所不支持多條符合條件的數值,如下圖:

 

Python解法:

for i in range(len(nums)-1):
             for j in range(i+1,len(nums)):
                 if nums[i]+nums[j] == target:
                     return [i,j]
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章