題目來源:力扣(LeetCode) 連接:https://leetcode-cn.com/problems/4sumjavascript
給定一個包含 n 個整數的數組 nums 和一個目標值 target,判斷 nums 中是否存在四個元素 a,b,c 和 d ,使得 a + b + c + d 的值與 target 相等?找出全部知足條件且不重複的四元組。java
注意:數組
答案中不能夠包含重複的四元組。性能
示例:優化
給定數組 nums = [1, 0, -1, 0, -2, 2],和 target = 0。 知足要求的四元組集合爲: [ [-1, 0, 0, 1], [-2, -1, 1, 2], [-2, 0, 0, 2] ]
題解:code
會了以前三數之和,這條題目就會變得簡單了,只須要在三數之和外面再套一層循環。排序
思路:ip
數組先進行排序。從小到大排序。循環最外層,至關於固定第一個數字,在進行一邊循環固定第二個數字,而後在剩下的數中找到和爲 target-nums[i]-nums[j]的兩個數字,最後組成相應的數組。內存
同時去重的操做要注意。第一層循環的時候,當nums[i] > Math.abs(target) || nums[i] === nums[i-1]直接進入下一層循環
等等。不過我這個方法雖然經過了,可是效率很低。leetcode
var fourSum = function(nums, target){ nums = nums.sort((a,b)=>{return a-b}); let arr = []; if(!nums || nums.length<4)return arr; for(let i = 0; i < nums.length-3;i++){ if(nums[i] > Math.abs(target) || nums[i] === nums[i-1])continue; for(let j = i + 1; j < nums.length-2;j++){ if(nums[j] === nums[j-1] && (j-1) !== i)continue; let sum = target - nums[i] - nums[j]; let left = j+1,right = nums.length-1; while(left < right){ if(nums[left]+nums[right]>sum){ do{--right}while(nums[right]===nums[right+1]) }else if(nums[left]+nums[right] < sum){ do{++left}while(nums[left]===nums[left-1]); }else{ let newArr = [nums[i],nums[j],nums[left],nums[right]]; arr.push(newArr); do{--right}while(nums[right]===nums[right+1]); do{++left}while(nums[left]===nums[left-1]) } } } } return arr; };
執行結果: 執行用時 :124 ms, 在全部 javascript 提交中擊敗了85.26% 的用戶 內存消耗 :36.5 MB, 在全部 javascript 提交中擊敗了81.40%的用戶
而後進行優化,除去一些狀況以後
/** * @param {number[]} nums * @param {number} target * @return {number[][]} */ var fourSum = function(nums, target) { nums = nums.sort((a,b)=>{return a-b}); let arr = [],n = nums.length; if(!nums || nums.length<4)return arr; for(let i = 0; i < nums.length-3;i++){ if(nums[i] > Math.abs(target) || nums[i] === nums[i-1])continue; let target1 = target - nums[i]; if (nums[i + 1] + nums[i + 2] + nums[i + 3] > target1) break; if (nums[n - 3] + nums[n - 2] + nums[n - 1] < target1) continue; for(let j = i + 1; j < nums.length-2;j++){ if(nums[j] === nums[j-1] && j>i+1)continue; let sum = target1 - nums[j]; if (nums[j + 1] + nums[j + 2] > sum) break; if (nums[n - 2] + nums[n - 1 < sum]) continue; let left = j+1,right = nums.length-1; while(left < right){ if(nums[left]+nums[right]>sum){ right--; }else if(nums[left]+nums[right] < sum){ left++; }else{ let newArr = [nums[i],nums[j],nums[left],nums[right]]; arr.push(newArr); do{--right}while(nums[right]===nums[right+1]); do{++left}while(nums[left]===nums[left-1]) } } } } return arr; }
執行結果 執行用時 :96 ms, 在全部 javascript 提交中擊敗了99.20% 的用戶 內存消耗 :36.2 MB, 在全部 javascript 提交中擊敗了94.57%的用戶
就是感受代碼量變大了不少,可是性能方面還行。