刷 LeetCode 跟著這個系列就行啦----兩數之和

三斤和他的喵發表於2020-04-08

前言

工作一段時間之後,最大的感覺就是演算法好像沒什麼用,確實不會演算法也能勝任平常的工作,但是總覺得缺了點什麼,所以最近抽空複習了以前刷的 Leetcode,希望在這裡找到一群志同道合的人:bowtie:

兩數之和(Two Sum)

這是 LeetCode 的第一題,總體來說是比較簡單的,題幹如下:

給定一個整數陣列 nums 和一個目標值 target,請你在該陣列中找出和為目標值的那兩個整數,並返回他們的陣列下標。
示例:
    給定 nums = [2, 7, 11, 15],target = 9
    因為 nums[0] + nums[1] = 2 + 7 = 9
    所以返回 [0, 1]

簡單來說就是給定一個陣列,讓你找到和為 target 的兩個元素的索引。

解題思路

主要有 兩種解題思路

  • 暴力解法:兩層迴圈,外部迴圈的當前值與target的差值為內層迴圈需要定位的值。時間複雜度O(n^2),空間複雜度O(1)。
  • 優雅解法:一層迴圈,引入map,迴圈的當前值與target的差值在map中查詢,如果不存在則將當前值作為鍵放入map,值的索引作為map的值,存在就返回。時間複雜度O(n),空間複雜度O(n)。相比於暴力解法其實就是空間換時間。

程式碼實現

暴力解法GO實現:

/*暴力解法
思路:
    兩層迴圈,外部迴圈的當前值與target的差值為內層迴圈需要定位的值
*/
func twoSumForce(nums []int, target int) (result []int) {
    length := len(nums)
    if length <= 0 {
        return
    }
    for i := 0; i < length; i++ {
        for j := i + 1; j < length; j++ {
            // 找到了就直接返回
            if nums[j] + nums[i] == target {
                return []int{i,j}
            }
        }
    }
    return
}

優雅解法GO實現:

/*優雅解法
思路:
    一層迴圈,引入map,迴圈的當前值與target的差值在map中查詢,如果不存在則將當前值作為鍵放入map,值的索引作為map的值。
*/
func twoSumElegant(nums []int, target int) (result []int) {
    if len(nums) <= 0 {
        return
    }
    var m = make(map[int]int)
    for index, value := range nums {
        if v, ok := m[target-value]; !ok {
            // 將當前值和索引放入map中
            m[value] = index
        } else {
            return []int{index,v}
        }
    }
    return
}

總結

每天進步一點點,加油!
完整程式碼:https://github.com/wx-satellite/go-leetcod...

本作品採用《CC 協議》,轉載必須註明作者和本文連結

三斤

相關文章