Leetcode --- 三数/四数问题

简介: Leetcode --- 三数/四数问题

1.三数之和(15 - 中)



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


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


示例 :

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


思路:先排序方便使用双指针查找元素,核心思想是:遍历数组,先用0 - 当前值作为sum,再使用双指针查找剩余两个元素。


注意


  • 对于含有重复元素的情况直接略过,否则会超时。
  • 双指针的左边界为i + 1


代码实现:

public List<List<Integer>> threeSum(int[] nums) {
    Arrays.sort(nums);
    List<List<Integer>> ans = new ArrayList<>();
    for (int i = 0; i < nums.length - 2; ++i) {
        if (i == 0 || (i > 0 && nums[i] != nums[i - 1])) {
            int l = i + 1, r = nums.length - 1;
            while (l < r) {
                int sum = nums[i] + nums[l] + nums[r];
                if (sum == 0) {
                    // 存储结果,数组转集合
                    ans.add(Arrays.asList(nums[i], nums[l], nums[r]));
                    // 防止相同的元素加入集合
                    while (l < r && nums[l] == nums[l + 1]) l++; 
                    while (l < r && nums[r] == nums[r - 1]) r--;
                    l++;
                    r--;
                } else if (sum > 0) {
                    r--;
                } else {
                    l++;
                }
            }
        }
    }
    return ans;
}


2.最接近的三数之和(16 - 中)



题目描述:给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。


示例 :

输入:nums = [-1,2,1,-4], target = 1
输出:2
解释:与 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。


思路:与上题三数之和基本类似的解法,遍历数组,更新三数之和与目标值最小的差值。


注意:当三数之和等于目标值可以直接返回(最近)。


代码实现:

public int threeSumClosest(int[] nums, int target) {
    Arrays.sort(nums);
    int closestNum = nums[0] + nums[1] + nums[2];
    for (int i = 0; i < nums.length - 2; ++i) {
        int l = i + 1, r = nums.length - 1;
        while (l < r){
            int threeSum = nums[l] + nums[r] + nums[i];
            if (Math.abs(threeSum - target) < Math.abs(closestNum - target)) {
                closestNum = threeSum;
            }
            if (threeSum > target) {
                r--;
            } else if (threeSum < target) {
                l++;
            } else {
                return target; 
            }
        }
    }
    return closestNum;
}


3.较小的三数之和(259 - 中)



题目描述:给定一个长度为 n 的整数数组和一个目标值 target,寻找能够使条件 nums[i] + nums[j] + nums[k] < target 成立的三元组 i, j, k 个数(0 <= i < j < k < n)。


示例 :

输入: nums = [-2,0,1,3], target = 2
输出: 2 
解释: 因为一共有两个三元组满足累加和小于 2:
     [-2,0,1]
     [-2,0,3]


思路:殊途同归,本题寻找小于目标值的三元组的个数。注意:若此时右边界组成的三元组符合条件,那么之前的位置都符合条件。


代码实现:

public int threeSumSmaller(int[] nums, int target) {
    Arrays.sort(nums);
    int ans = 0;
    for (int i = 0; i < nums.length - 2; ++i) {
        int l = i + 1, r = nums.length - 1;
        while (l < r){
            int sum = nums[l] + nums[r] + nums[i];
            if (sum >= target) {
                r--;
            } else {
                ans += r - l;
                l++;
            }
        }
    }
    return ans;
}


4.四数之和(18 - 中)



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


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


示例 :

输入:nums = [1,0,-1,0,-2,2], target = 0
输出:[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]


思路:本质不变,在三数之和基础上又加了一层循环。还是注意对于重复值的剔除(跳过)。


代码实现:

public List<List<Integer>> fourSum(int[] nums, int target) {
    Arrays.sort(nums);
    List<List<Integer>> ans = new ArrayList<>();
    int n = nums.length;
    for (int i = 0; i < n - 3; i++) {
        if (i > 0 && nums[i] == nums[i - 1]) continue;
        for (int j = i + 1; j < n - 2; j++) {
            if (j > i + 1 && nums[j] == nums[j - 1]) continue;
            int l = j + 1, r = n - 1;
            while (l < r) {
                int sum = nums[i] + nums[j] + nums[l] + nums[r];
                if (sum == target) {
                    ans.add(Arrays.asList(nums[i], nums[j], nums[l], nums[r]));
                    while (l < r && nums[l] == nums[l + 1]) l++;
                    while (l < r && nums[r] == nums[r - 1]) r--;
                    l++;
                    r--;
                } else if (sum > target) {
                    r--;
                } else {
                    l++;
                }
            }
        }
    }
    return ans;
}


相关文章
|
6天前
[leetcode] 四数之和 M
[leetcode] 四数之和 M
|
6天前
|
Java 测试技术 C++
leetcode-18:四数之和
leetcode-18:四数之和
26 0
|
6天前
|
Java C++ Python
leetcode-15:三数之和
leetcode-15:三数之和
22 0
|
11月前
|
算法 安全 Swift
LeetCode - #18 四数之和
不积跬步,无以至千里;不积小流,无以成江海,Swift社区 伴你前行。如果大家有建议和意见欢迎在文末留言,我们会尽力满足大家的需求。
|
11月前
leetcode:18.四数之和
这题和前面的一道三数之和类似,解题的思路都一样,这里直接选取两个基准就可以了,然后循环出所有的组合进行判断,如果正好相等那么就加入Set集合中。
42 0
|
11月前
|
算法 测试技术
leetcode:15.三数之和
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
59 0
|
11月前
|
存储
力扣 -- 15.三数之和
力扣 -- 15.三数之和
|
算法
LeetCode每日1题--三数之和
LeetCode每日1题--三数之和
59 0
|
算法
LeetCode每日1题--四数之和
LeetCode每日1题--四数之和
59 0
|
Java C++ Python
【LeetCode】 15. 三数之和
第15题. 三数之和
57 0
【LeetCode】 15.  三数之和