leetcode不按期刷題---18. 四數之和

題目來源:力扣(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%的用戶

就是感受代碼量變大了不少,可是性能方面還行。

相關文章
相關標籤/搜索