1. 程式人生 > 實用技巧 >雙指標演算法題--- 三數之和

雙指標演算法題--- 三數之和

題目描述:

給你一個包含 n 個整數的陣列 nums,判斷 nums 中是否存在三個元素 a,b,c ,使得 a + b + c = 0 ?請你找出所有滿足條件且不重複的三元組。

注意:答案中不可以包含重複的三元組。

示例:

給定陣列 nums = [-1, 0, 1, 2, -1, -4],

滿足要求的三元組集合為:[ [-1, 0, 1], [-1, -1, 2] ]

解題思路:

雙指標

這道題是典型的雙指標應用,雙指標和此題的面試出鏡率都很高,建議掌握!

這道題的演算法思路如下:

  1. 特殊判斷,對於陣列長度 n,如果陣列為 null 或者陣列長度小於 3,直接返回 []。
  2. 對陣列進行排序(排序是必須的,否則無法判斷及指標移動)。
  3. 遍歷排序後陣列:
    • 若 nums[i]>0:因為已經排序好,所以後面不可能有三個數加和等於 0,直接返回結果。
    • 對於重複元素:跳過,避免出現重複解
    • 令左指標 L=i+1,右指標 R=n-1,當 L<R 時,執行迴圈:
      • 當 nums[i]+nums[L]+nums[R]==0,滿足要求將元素新增到結果集中,執行迴圈,判斷左界和右界是否和下一位置重複,去除重複解。並同時將 L,R 移到下一位置,尋找新的解
      • 若和大於 0,說明 nums[R] 太大,R 左移
      • 若和小於 0,說明 nums[L] 太小,L 右移

程式碼實現

//go
func threeSum(nums []int) [][]int {
 var result [][]int // 記錄返回結果
 length := len(nums) // 獲取陣列長度
 if nums == nil || length < 3 {
  return result
 }
 sort.Ints(nums) // 排序
 for i := 0; i < length; i++ {
  // 如果當前數字大於0,則三數之和一定大於0,所以結束迴圈
  if nums[i] > 0 {
   break
  }
  // 去重,下標 i 的數
  if i > 0 && nums[i] == nums[i-1] {
   continue
  }
  l := i + 1
  r := length - 1
  for {
   if l >= r {
    break
   }
   sum := nums[i] + nums[l] + nums[r]
   if sum > 0 {
    r --
    continue
   }
   if sum < 0 {
    l ++
    continue
   }
   if sum == 0 {
    result = append(result, []int{nums[i], nums[l], nums[r]})
    // 去重,下標 l
    for l < r && nums[l] == nums[l+1] {
     l++
    }
    // 去重,下標 r
    for l < r && nums[r] == nums[r-1] {
     r--
    }
    l++
    r--
   }
  }
 }
 return result
}

  地址:https://mp.weixin.qq.com/s/2czJg7xt004ugRE9NfhSwg