3합/3합

6091 단어 javascriptleetcode
정수 배열 nums가 주어지면 i != j, i != k, j != k 및 nums[i] + 숫자[j] + 숫자[k] == 0.

솔루션 세트에는 중복된 삼중항이 포함되어서는 안 됩니다.

예 1:

입력: 숫자 = [-1,0,1,2,-1,-4]
출력: [[-1,-1,2],[-1,0,1]]
설명:
숫자[0] + 숫자[1] + 숫자[2] = (-1) + 0 + 1 = 0.
숫자[1] + 숫자[2] + 숫자[4] = 0 + 1 + (-1) = 0.
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0.
뚜렷한 삼중 항은 [-1,0,1]과 [-1,-1,2]입니다.
출력의 순서와 셋의 순서는 중요하지 않습니다.

const threeSum = (arr) => {
  let nums = [...arr].sort();
  const res = []; 

  for (let i = 0; i < nums.length - 2; i++) {
    // We can do - 2 cause our other 2 pointers will take care of the last 2 nums
    // Initialize 2 pointers for innerloop which will give sum 
    let l = i + 1,
      r = nums.length - 1;

    if (nums[i] === nums[i - 1]) continue; // check for dupes on the outter loop, skip if nums[i] is a num we already saw

    while (l < r) {
      let total = nums[i] + nums[l] + nums[r]; // Add up all our current pointer values

      if (total === 0) {
        res.push([nums[i], nums[l], nums[r]]); // Add if we found a solution

        while (nums[l] === nums[l + 1]) l++; // This skips dupes in the nums array
        while (nums[r] === nums[r - 1]) r--; // This skips dupes in the nums array

        // We need to do one extra move after skipping the dupes or else we will still be on a dupe
        l++;
        r--;
      } else if (total > 0) {
        // If its greater than 0, then we can move the r pointer down
        r--;
      } else {
        // its less than 0 and we need to move the left pointer up
        l++;
      }
    }
  }

  return res;
};


좋은 웹페이지 즐겨찾기