代码随想录算法训练营第七天| 454.四数相加、383.赎金信、15. 三数之和、 18. 四数之和

天真的和感伤的

已于 2023-08-08 23:22:37 修改

阅读量84

点赞数

文章标签： java 算法

于 2023-07-31 16:08:03 首次发布

本文链接：https://blog.csdn.net/Viking101/article/details/132024509

版权

454. 四数相加 II

参考文档：代码随想录
参考视频：学透哈希表，map使用有技巧！LeetCode：454.四数相加II_哔哩哔哩_bilibili

给你四个整数数组 nums1、nums2、nums3 和 nums4 ，数组长度都是 n ，请你计算有多少个元组 (i, j, k, l) 能满足：

0 <= i, j, k, l < n
nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0

示例 1：

输入：nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2]

输出：2
解释：
两个元组如下：
1. (0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0
2. (1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0

示例 2：

输入：nums1 = [0], nums2 = [0], nums3 = [0], nums4 = [0]
输出：1

提示：

n == nums1.length
n == nums2.length
n == nums3.length
n == nums4.length
1 <= n <= 200
-228 <= nums1[i], nums2[i], nums3[i], nums4[i] <= 228

思路：
两个数组一对求和时间复杂度最低，先求出nums1和nums2中所有的和的情况，并且统计次数，分别作为key，value存入到HashMap中，再求出0减去nums3和nums4值，判断该值在map中是否存在，若存在将该值对应的value取出，相加得出最后的结果。

完整代码(java)

package leetcode2;

import java.util.HashMap;
import java.util.Map;

/**
 * @description:
 * @author: lxc
 * @date: 2023/7/31 15:16
 */
public class Question454 {

    public static void main(String[] args) {
        
        int[] nums1 = {1,2};
        int[] nums2 = {-2,-1};
        int[] nums3 = {-1,2};
        int[] nums4 = {0,2};

        int i = fourSumCount(nums1, nums2, nums3, nums4);
        System.out.println(i);
    }

    private static int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {

        Map<Integer,Integer> map = new HashMap<>();
        int res = 0;

        // 先统计两个数之和，并同时统计出现的次数，放入map中
        for (int i : nums1) {
            for (int j : nums2) {
                int sum = i + j;
                map.put(sum,map.getOrDefault(sum,0) + 1);
            }
        }
        
        // 再统计剩余的两个元素的和，在map中寻找是否存在相加为0的情况，同时记录次数
        for (int i : nums3) {
            for (int j : nums4) {
                res += map.getOrDefault(0 - i - j,0);
            }
        }
        return res;
    }
}

383. 赎金信

参考文档：代码随想录

给你两个字符串：ransomNote 和 magazine ，判断 ransomNote 能不能由 magazine 里面的字符构成。

如果可以，返回 true ；否则返回 false 。

magazine 中的每个字符只能在 ransomNote 中使用一次。

示例 1：

输入：ransomNote = "a", magazine = "b"
输出：false

示例 2：

输入：ransomNote = "aa", magazine = "ab"
输出：false

示例 3：

输入：ransomNote = "aa", magazine = "aab"
输出：true

提示：

1 <= ransomNote.length, magazine.length <= 105
ransomNote 和 magazine 由小写英文字母组成

思路: 因为字母的数量是有限的，故可以用空间换时间，利用数据处理这道题目。先记录magazine中所有字母的个数，按位置存在数组中，再遍历ransomNode，将对应的字母从数组中减去，如果出现复数，则说明magazine中字母的字数不够，无法组成赎金信，返回false。

完整代码(java)

package leetcode2;

/**
 * @description:
 * @author: lxc
 * @date: 2023/7/31 16:21
 */
public class Question383 {

    public static void main(String[] args) {

        String a = "aa";
        String b = "aab";

        boolean res = canConstruct(a, b);
        System.out.println(res);

    }

    // 因为字母的数量有限，所以使用数组
    private static boolean canConstruct(String ransomNote, String magazine) {

        // 判断magazine是否闭ransomNote大，如果不是返回false
        if (magazine.length() < ransomNote.length()){
            return false;
        }

        int[] chars = new int[26];

        for (char c : magazine.toCharArray()) {
            chars[c - 'a'] += 1;
        }

        for (char c : ransomNote.toCharArray()) {
            chars[c - 'a'] -= 1;
        }

        // 判断chars中是否有小于0的元素，如果有，则说明magazine中的字母数量不够，返回false
        for (int c : chars) {
            if (c < 0){
                return false;
            }
        }

        return true;
    }
}

15. 三数之和

参考文档：代码随想录
参考视频：梦破碎的地方！| LeetCode：15.三数之和_哔哩哔哩_bilibili

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

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

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
解释：
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意，输出的顺序和三元组的顺序并不重要。

示例 2：

输入：nums = [0,1,1]
输出：[]
解释：唯一可能的三元组和不为 0 。

示例 3：

输入：nums = [0,0,0]
输出：[[0,0,0]]
解释：唯一可能的三元组和为 0 。

思路：利用双指针法。对数组进行排序，再遍历数组，一个指针指向当前遍历的元素下一个位置，另一个指针指向数组中的最后一个元素。先用i > 0 && nums[i] == nums[i - 1]对结果数组中的第一个元素进行去重，如果相等则跳过此次循环，将遍历到的当前元素与两个指针所指向的元素相加，得到的结果与0做比较，大于0，右指针左移，小于0，左指针右移，当right > left时跳出循环。找到三元组后再对结果数组的第二位和第三位进行去重，如果nums[right] == nums[right - 1]，nums[left] == nums[left - 1]，说明有重复，right--或left++。

完整代码(java)

package leetcode2;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @description:
 * @author: lxc
 * @date: 2023/8/7 17:50
 */
public class Question15 {

    public static void main(String[] args) {

        int[] nums = new int[]{-1,0,1,2,-1,-4};
        List<List<Integer>> lists = threeSum(nums);
        System.out.println(lists);

    }

    private static List<List<Integer>> threeSum(int[] nums) {

        List<List<Integer>> res = new ArrayList<>();

        if (nums == null && nums.length < 3){
            return res;
        }

        // 对数组进行排序
        Arrays.sort(nums);

        // 对数组进行遍历
        for (int i = 0; i < nums.length; i++) {

            // 如果数组排序后当前数大于0，说明三数之和一定大于0，循环结束
            if (nums[i] > 0){
                break;
            }
            // 对结果数组第一位数进行去重
            if (i > 0 && nums[i] == nums[i - 1]){
                continue;
            }

            int left = i + 1;
            int right = nums.length - 1;

            while (left < right){
                int sum  = nums[i] + nums[left] + nums[right];
                if (sum == 0){
                    res.add(Arrays.asList(nums[i],nums[left],nums[right]));
                    // 对结果数组第二位进行去重
                    while (left < right && nums[left] == nums[left + 1]){
                        left++;
                    }
                    // 对结果数组第三位进行去重
                    while (left < right && nums[right] == nums[right - 1]){
                        right--;
                    }
                    left++;
                    right--;
                }else if(sum < 0){
                    left++;
                }else if(sum > 0){
                    right--;
                }
            }
        }
        return res;
    }
}

18. 四数之和

参考文档：代码随想录
参考视频：难在去重和剪枝！| LeetCode：18. 四数之和_哔哩哔哩_bilibili

给你一个由 n 个整数组成的数组 nums ，和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] （若两个四元组元素一一对应，则认为两个四元组重复）：

0 <= a, b, c, d < n
a、b、c 和 d 互不相同
nums[a] + nums[b] + nums[c] + nums[d] == target

你可以按任意顺序返回答案。

示例 1：

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

输出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

示例 2：

输入：nums = [2,2,2,2,2], target = 8

输出：[[2,2,2,2]]

提示：

1 <= nums.length <= 200
-109 <= nums[i] <= 109
-109 <= target <= 109

思路：采用双指针法，类似与三数之和的思路。需要注意的细节是剪枝和去重。利用两层for循环，去遍历数组，左右指针往中间移。

完整代码(java)

package leetcode2;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @description:
 * @author: lxc
 * @date: 2023/8/8 19:41
 */
public class Question18 {

    public static void main(String[] args) {

        int[] nums = new int[]{1,0,-1,0,-2,2};
        int target = 0;
        List<List<Integer>> lists = fourSum(nums, target);
        System.out.println(lists);
    }

    private static List<List<Integer>> fourSum(int[] nums, int target) {

        List<List<Integer>> result = new ArrayList<>();
        Arrays.sort(nums);

        for (int i = 0; i < nums.length; i++) {

            // nums[i] > target 直接返回，剪枝操作
            if (nums[i] > 0 && nums[i] > target) {
                return result;
            }

            // 对nums[i]去重
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }

            for (int j = i + 1; j < nums.length; j++) {

                // 剪枝
                if (nums[i] + nums[j] > target && nums[i] + nums[j] > 0) {
                    break;
                }

                // 对nums[j]去重
                if (j > i + 1 && nums[j - 1] == nums[j]) {
                    continue;
                }

                int left = j + 1;
                int right = nums.length - 1;
                while (right > left) {
                    long sum = (long) nums[i] + nums[j] + nums[left] + nums[right];
                    if (sum > target) {
                        right--;
                    } else if (sum < target) {
                        left++;
                    } else {
                        result.add(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));
                        // 对nums[left] 和 nums[right]进行去重
                        while (right > left && nums[right] == nums[right - 1]){
                            right--;
                        }
                        while (right > left && nums[left] == nums[left + 1]){
                            left++;
                        }

                        left++;
                        right--;
                    }
                }
            }
        }
        return result;
    }
}