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

LeetCode 454 四数相加
题目链接：四数相加

思路：四数相加最暴力的做法是四层循环，但其实可以拆解成两个双层循环，这样时间复杂度为二次。

package HashTable;

import java.util.HashMap;

/**
 * @author rpstart
 * @create 2023-07-04 22:42
 */
public class fourSumCount454 {
    public static void main(String[] args) {

    }

    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        int res = 0;
        int sum = 0;
        HashMap<Integer, Integer> map = new HashMap<>();
        for (int i : nums1) {
            for (int j : nums2) {
                sum = i + j;
                map.put(sum, map.getOrDefault(sum, 0) + 1);
            }
        }

        for (int i : nums3) {
            for (int j : nums4) {
                res += map.getOrDefault(-i - j, 0);
            }
        }
        return res;
    }
}

LeetCode 383 赎金信
题目链接：赎金信

思路：在处理字符串问题时，同时必须使用哈希结构，一般使用数组就可以处理问题，没必要使用map。

package HashTable;

import java.util.HashMap;

/**
 * @author rpstart
 * @create 2023-07-04 22:56
 */
public class canConstruct383 {
    public static void main(String[] args) {
        String s = "aa";
        String t = "aab";
        boolean res = canConstruct(s, t);
        System.out.println(res);
    }

    public static boolean canConstruct(String ransomNote, String magazine) {
        if (ransomNote.length() > magazine.length()) {
            return false;
        }

        int[] record = new int[26];

        for (int i = 0; i < magazine.length(); i++) {
            record[magazine.charAt(i) - 'a']++;
        }
        for (int i = 0; i < ransomNote.length(); i++) {
            record[ransomNote.charAt(i) - 'a']--;
        }

        for (int i : record) {
            if (i < 0) {
                return false;
            }
        }
        return true;
    }
}

LeetCode 15 三数之和
题目链接：三数之和

思路：关键点在于1、想到用双指针减少一层循环；2、双指针能用的前提是数组要先进行排序，但本题的时间复杂度至少与排序复杂度持平，因此直接使用排序没问题。3、如何对三个元素去重。

package HashTable;

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

/**
 * @author rpstart
 * @create 2023-07-04 23:13
 */
public class threeSum15 {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        Arrays.sort(nums);
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] > 0) {
                return res;
            }
            if (i != 0 && nums[i] == nums[i - 1]) {
                continue;
            }
            int left = i + 1;
            int right = nums.length - 1;
            while (right > left) {
                int sum = nums[i] + nums[left] + nums[right];
                if (sum > 0) {
                    right--;
                } else if (sum < 0) {
                    left++;
                } else {
                    res.add(Arrays.asList(nums[i], nums[left], nums[right]));
                    while (right > left && nums[right] == nums[right - 1]) {
                        right--;
                    }
                    while (right > left && nums[left] == nums[left + 1]) {
                        left++;
                    }
                    right--;
                    left++;
                }
            }
        }
        return res;
    }
}

LeetCode 18 四数之和
题目链接：四数之和

思路：该题是在三数之和的基础上加了一层循环，但需注意的是去重条件的情况。

package HashTable;

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

/**
 * @author rpstart
 * @create 2023-07-04 23:36
 */
public class fourSum18 {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        List<List<Integer>> res = new ArrayList<>();
        Arrays.sort(nums);
        for (int i = 0; i < nums.length; i++) {
            if (nums[i] > 0 && nums[i] > target) {
                return res;
            }

            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }
            for (int j = i + 1; j < nums.length; 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 {
                        res.add(Arrays.asList(nums[i], nums[j], nums[left], nums[right]));
                        while (right > left && nums[right - 1] == nums[right]) {
                            right--;
                        }
                        while (right > left && nums[left + 1] == nums[left]) {
                            left++;
                        }
                        right--;
                        left++;
                    }
                }
            }
        }
        return res;
    }
}