美文网首页算法
leetcode 第18题 四数之和

leetcode 第18题 四数之和

作者: 嘿晴天 | 来源:发表于2019-05-25 14:15 被阅读0次

    这道题是三数之和的拓展题,https://www.jianshu.com/writer#/notebooks/2661140/notes/46891047

    四数之和 题目

    给定一个包含 n 个整数的数组 nums 和一个目标值 target,判断 nums 中是否存在四个元素 a,b,c 和 d ,使得 a + b + c + d 的值与 target 相等?找出所有满足条件且不重复的四元组。

    注意:

    答案中不可以包含重复的四元组。

    示例:

    给定数组 nums = [1, 0, -1, 0, -2, 2],和 target = 0。

    满足要求的四元组集合为:
    [
    [-1, 0, 0, 1],
    [-2, -1, 1, 2],
    [-2, 0, 0, 2]
    ]

    解题思路: 可以参照三数之和的解题思路,不过这次需要固定两个数,也其实是三数之和外层多一个循环

    代码如下

    /**
     * @param {number[]} nums
     * @param {number} target
     * @return {number[][]}
     */
    var fourSum = function(nums, target) {
        var number = [];
    
        if (nums.length < 4)
        {
            return number;
        }
            
        nums = nums.sort(sortNumber);
        
        for (var i = 0; i < nums.length - 3;  i++)
        {
             if (i > 0 && nums[i] == nums[i -1])
             {
                continue;
             }
             
            for (var j = i + 1; j < nums.length - 2; j++)
                {
                    var k = j+1; 
                    var l = nums.length -1;
                    if (j > i +1 && nums[j] == nums[j -1])
                    {
                        continue;
                    }
                        
                    while (k < l)
                    {
                        if (k > j +1 && nums[k] == nums[k -1])
                        {
                            k++;
                            continue;
                        }
                        
                        if (l < nums.length - 1 && nums[l] == nums[l+1])
                        {
                            l--;
                            continue;
                        }
                           
                        var sum = nums[i] + nums[j] + nums[k] + nums[l];
                        if (sum == target)
                        {
                            number.push([nums[i], nums[j], nums[k], nums[l]])
                            k++;
                            l--;
                            
                        } else if (sum > target) {
                            l--;
                        } else if (sum < target) {
                            k++;
                        }
                    }
                }
        }
        return number;
    };
    
    function sortNumber(a,b)
    {
        return a - b
    }
    
    
    

    相关文章

      网友评论

        本文标题:leetcode 第18题 四数之和

        本文链接:https://www.haomeiwen.com/subject/vyiazqtx.html