1. 程式人生 > >LeetCode——中級演算法——陣列和字串——三數之和

LeetCode——中級演算法——陣列和字串——三數之和

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

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

例如, 給定陣列 nums = [-1, 0, 1, 2, -1, -4],
滿足要求的三元組集合為:
[
[-1, 0, 1],
[-1, -1, 2]
]


思路

如果這題用暴力求解,三層迴圈,不僅慢,而且會出現重複的三元組。
而三元組內部允許出現重複元素,例如 [-1, -1, 2],因此也不能使用set集合。
因此想到,先固定一個數,然後在用雙指標在其餘數中尋找和為其相反數的兩個數

,但怎樣才能避免出現重複的三元組呢?
答案是 先將陣列排序。將陣列排序後,重複的數直接跳過。
所以第一步,排序:

nums.sort((a,b) => a - b);

注意預設的排序sort()會把 -2 排在 -4 前面,造成 [ -2 , -3 , -4, 0, 2, 3, 3, 4] 這種情況。
先固定一個負數,然後在後面的數中找兩個數。
使用k作為固定數的索引,i、j 作為雙指標從 k + 1 和 陣列末尾開始。target作為nums[k]的相反數

let target = 0 - nums[k]
let i = k + 1;
let j = nums.length -
1;

當 i < j 時迴圈

如果i、j所指的數相加等於target,說明可以組成三元組,加入result。
然後把 i 往後移到下一個不相同的數,把 j 往前移到下一個不相同的數。
如果小了,移 i; 大了就移 j。

總程式碼:

/**
 * @param {number[]} nums
 * @return {number[][]}
 */
var threeSum = function(nums) {
  nums.sort((a,b) => a - b);
  let result = [];
  for (let k = 0; k < nums.length;
k ++) { if (nums[k] > 0) break; if (nums[k] === nums[k-1]) continue; let target = 0 - nums[k]; let i = k + 1, j = nums.length - 1; while (i < j) { if (nums[i] + nums[j] === target) { result.push([nums[k], nums[i], nums[j]]) while (i < j && nums[i] === nums[i + 1]) { i ++; } while (i < j && nums[j] === nums[j - 1]) { j --; } ++i; --j; } else if (nums[i] + nums[j] < target) { ++i; } else { --j; } } } return result; };