给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。

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

    示例 1:

    输入:nums = [-1,0,1,2,-1,-4]
    输出:[[-1,-1,2],[-1,0,1]]
    示例 2:

    输入:nums = []
    输出:[]
    示例 3:

    输入:nums = [0]
    输出:[]

    1. /**
    2. * @param {number[]} nums
    3. * @return {number[][]}
    4. */
    5. var threeSum = function (nums) {
    6. nums.sort((a, b) => a - b);
    7. const nSumTarget = (nums, n, start, target) => {
    8. let length = nums.length;
    9. let res = [];
    10. // 至少是求两数之和,且大小不小于N 不然没意义
    11. if (n < 2 || length < n) return res;
    12. // 基础2数之和
    13. if (n === 2) {
    14. // 双指针
    15. let l = start,
    16. r = length - 1;
    17. while (l < r) {
    18. let sum = nums[l] + nums[r];
    19. let _left = nums[l],
    20. _right = nums[r];
    21. if (sum < target) {
    22. while (l < r && nums[l] === _left) l++;
    23. } else if (sum > target) {
    24. while (l < r && nums[r] === _right) r--;
    25. } else {
    26. res.push([_left, _right])
    27. while (l < r && nums[l] === _left) l++;
    28. while (l < r && nums[r] === _right) r--;
    29. }
    30. }
    31. } else {
    32. // 大于2时就递归计算 (N-1)Sum 的结果
    33. for (let i = start; i < length; i += 1) {
    34. let sub = nSumTarget(nums, n - 1, i + 1, target - nums[i]);
    35. for (let arr of sub) {
    36. // (n-1)Sum 加上 nums[i] 就是 nSum
    37. res.push([...arr, nums[i]]);
    38. }
    39. while (i < length - 1 && nums[i] === nums[i + 1]) i++;
    40. }
    41. }
    42. return res;
    43. }
    44. return nSumTarget(nums, 3, 0, 0)
    45. };

    image.png