LEETCODE - 0001 - 兩數之和

原文連接

給定一個整數數組和一個目標值，找出數組中和爲目標值的兩個數。

你能夠假設每一個輸入只對應一種答案，且一樣的元素不能被重複利用。

示例:

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

由於 nums[0] + nums[1] = 2 + 7 = 9 因此返回 [0, 1]

來源：力扣（LeetCode） 連接：leetcode-cn.com/problems/tw… 著做權歸領釦網絡全部。商業轉載請聯繫官方受權，非商業轉載請註明出處。

解題思路

1. 遍歷輸入的數組，計算當前數字須要的另外一個數字 n2 = target - n1。
2. 從額外的map中尋找 n2
   • 若是能找到，則當前數字n1的下標i1和map中存儲的下標m[n2]組成答案返回
   • 若是未找到，則記錄當前數字的值和下標m[val] = idx

完整代碼

Golang

func TwoSum(nums []int, target int) []int {
    m := map[int]int{}

    for i1, n1 := range nums {
        n2 := target - n1
        i2, ok := m[n2]
        if ok {
            return []int{i2, i1}
        }

        m[n1] = i1
    }

    return nil
}
複製代碼

Common Lisp

遞歸

(defun two-sum (lst tar)
  (two-sum-helper lst tar 0
                  (find-num-index (cdr lst) (- tar (car lst)) 1)))

(defun two-sum-helper (lst tar m n)
  (if (null lst) nil
      (if (numberp n)
          (list m n)
          (two-sum-helper (cdr lst) tar (+ m 1)
                          (find-num-index (cdr lst) (- tar (car lst)) (+ m 2))))))

(defun find-num-index(lst tar n)
  (if (null lst) nil
      (if (eql tar (car lst))
          n
          (find-num-index (cdr lst) tar (+ n 1)))))
複製代碼

非遞歸

(defun two-sum (lst tar)
  (let ((ht (make-hash-table))
        (l (length lst))
        (n nil))
    (do ((i 0 (+ i 1)))
        ((> i l) nil)
      (setf n (gethash (- tar (car lst)) ht))
      (if (numberp n)
          (return-from two-sum (list n i)))
      (setf (gethash (car lst) ht) i)
      (pop lst))))
複製代碼