给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
1/** 2 * @param {number[]} nums 3 * @return {number[][]} 4 */ 5 6var threeSum = function (nums) { 7 // 去重到每个最多只剩两个(如果是0, 至多3个) 8 nums = unique(nums) 9 if (nums.length < 3) return []; 10 11 var r = []; 12 nums.sort((a, b) => a - b); 13 14 for (var i = 0; i < nums.length - 2; i++) { 15 var newNums = nums.slice(i + 1); 16 17 // a + b + c = 0 => a + b = -c 将三数之和转 化为 两数之和 18 var a = twoSum(newNums, nums[i]); 19 if (a.length !== 0) r = r.concat(a) 20 } 21 22 return depute(r) 23} 24 25// 筛选数据至每个数据出现不超过两次(0三次) 26/** 27 * @param {number[]} nums 28 * @return {number[]} 29 */ 30var unique = function(nums) { 31 // 出现次数不超过2次(如果是0那就3次) 32 var r = [], r1 = [] 33 for(var i=0; i<nums.length; i++) { 34 if(r.indexOf(nums[i]) === -1){ 35 r.push(nums[i]) 36 nums.splice(i, 1) 37 i-- 38 } 39 } 40 41 for(var i=0; i<nums.length; i++) { 42 if(r1.indexOf(nums[i]) === -1){ 43 r1.push(nums[i]) 44 nums.splice(i, 1) 45 i-- 46 } 47 } 48 49 var r = r.concat(r1) 50 51 // 如果还有 0 52 for(var i=0; i<nums.length; i++) { 53 if(nums.indexOf(0) !== -1) { 54 r.push(0) 55 break 56 } 57 } 58 59 return r 60} 61 62// 两数之和 63/** 64 * @param {number[]} numbers 65 * @param {number} target 66 * @return {number[]} 67 */ 68var twoSum = function (numbers, target) { 69 var r = [], l = numbers.length 70 71 for (var i = 0; i < l - 1;) { // i < l-1 而不是 i < l 72 if (numbers[i] + numbers[l - 1] < -target) { 73 i++ 74 continue 75 } 76 if (numbers[i] + numbers[l - 1] > -target) { 77 l-- 78 continue 79 } 80 if (numbers[i] + numbers[l - 1] === -target) { 81 var a = [target, numbers[i], numbers[l - 1]] 82 r.push(a) 83 i++ 84 l-- 85 } 86 } 87 88 return r 89} 90 91// 深度去重 92/** 93 * @param {number[]} a 94 * @return {number[]} 95 */ 96var depute = function (a) { 97 for (var i = 0; i < a.length; i++) { 98 a[i] = JSON.stringify(a[i]) 99 }100101 var newArr = new Set(a)102 newArr = Array.from(newArr)103104 for (var j = 0; j < newArr.length; j++) {105 newArr[j] = JSON.parse(newArr[j])106 }107108 return newArr109}
原文地址:https://www.cnblogs.com/rencoo/p/9822276.html
时间: 2024-11-09 23:49:26