代码随想录训练营二刷第七天 | 454.四数相加II 383. 赎金信 15. 三数之和 18. 四数之和
一、454.四数相加II
题目链接：https://leetcode.cn/problems/4sum-ii/
思路：求索引的组合，转化为两组，a+b = -(c+d)，map中键为a+b，值为出现的次数。然后去匹配c+d即可。
class Solution {public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {int sum = 0;HashMap<Integer, Integer> map = new HashMap<>();for (int i : nums1) {for (int j : nums2) {int temp = i + j;map.put(temp, map.getOrDefault(temp, 0) + 1);}}for (int i : nums3) {for (int j : nums4) {int temp = i + j;if (map.containsKey(-temp)) {sum += map.getOrDefault(-temp, 0);}}}return sum;}
}
二、383. 赎金信
题目链接：https://leetcode.cn/problems/ransom-note/
思路：字母经典数组操作。数组里如果存放r字符串要变量3次（如果使用一个变量计数就可以只遍历2次），如果存放m字符串只需要遍历2次。
class Solution {public boolean canConstruct(String ransomNote, String magazine) {int[] nums = new int[26];for (int i = 0; i < ransomNote.length(); i++) {nums[ransomNote.charAt(i)-'a']++;}for (int i = 0; i < magazine.length(); i++) {if (nums[magazine.charAt(i)-'a'] > 0) {nums[magazine.charAt(i)-'a']--;}}for (int i = 0; i < nums.length; i++) {if (nums[i] != 0) {return false;}}return true;}
}// =======================
public boolean canConstruct(String ransomNote, String magazine) {int[] nums = new int[26];int sum = 0;for (int i = 0; i < ransomNote.length(); i++) {nums[ransomNote.charAt(i)-'a']++;sum++;}for (int i = 0; i < magazine.length(); i++) {int k = magazine.charAt(i) - 'a';if (nums[k] > 0) {nums[k]--;sum--;}}return sum == 0;}
三、15. 三数之和
题目链接：https://leetcode.cn/problems/3sum/
思路：三数之和收集数组中元素，可以排序后，内层双指针遍历，去重，从a\b\c收集时考虑，和自己上一次收集的值一致时即跳过。
class Solution {public List<List<Integer>> threeSum(int[] nums) {Arrays.sort(nums);List<List<Integer>> list = new ArrayList<>();for (int i = 0; i < nums.length - 2; i++) {if (i > 0 && nums[i] == nums[i-1]) continue;int j = i+1, k = nums.length - 1;while (j < k) {int temp = nums[i] + nums[j] + nums[k];if (temp == 0) {list.add(Arrays.asList(nums[i], nums[j], nums[k]));while (j < k && nums[j] == nums[j+1]) j++;while (j < k && nums[k] == nums[k-1]) k--;j++;k--;}else if (temp > 0) {k--;} else {j++;}}}return list;}
}
四、18. 四数之和
题目链接：https://leetcode.cn/problems/4sum/
思路：注意剪枝，去重。
public List<List<Integer>> fourSum(int[] nums, int target) {Arrays.sort(nums);List<List<Integer>> list = new ArrayList<>();for (int i = 0; i < nums.length - 3; i++) {if (nums[i] > 0 && nums[i] > target) return list;if (i > 0 && nums[i] == nums[i-1]) continue;for (int j = i+1; j < nums.length -2; j++) {if (nums[j] + nums[i] > target && nums[i] >= 0) {break;}if (j > i+1 && nums[j] == nums[j-1]) continue;int k = j + 1, h = nums.length - 1;while (k < h) {long temp = (long) nums[i] + nums[j] + nums[k] + nums[h];if (temp < target) {k++;}else if (temp > target) {h--;} else {list.add(Arrays.asList(nums[i], nums[j], nums[k], nums[h]));while (k < h && nums[k] == nums[k+1]) k++;while (k < h && nums[h] == nums[h-1]) h--;k++;h--;}}}}return list;}
)
