代码随想录算法训练营第七天|454.四数相加||、383.赎金信、15.三数之和、18.四数之和
454.四数相加||
给你四个整数数组 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
解释:
两个元组如下:
- (0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0
- (1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0
题解:四个数组,那就两两求和,然后将结果放入map。注意键值对分别是什么,键是和,值是和出现的次数。
代码:
class Solution {
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
Map<Integer,Integer> map=new HashMap<>();
int res=0;
for(int i:nums1){
for(int j:nums2){
int sum=i+j;
map.put(sum,map.getOrDefault(sum,0)+1);
}
}
for(int i:nums3){
for(int j:nums4){
int sum=i+j;
res+=map.getOrDefault(0-sum,0);
}
}
return res;
}
}
383.赎金信
给你两个字符串:ransomNote
和 magazine
,判断 ransomNote
能不能由 magazine
里面的字符构成。
如果可以,返回 true
;否则返回 false
。
magazine
中的每个字符只能在 ransomNote
中使用一次。
示例 1:
输入:ransomNote = "a", magazine = "b"
输出:false
题解:这和上一题方法是一样的。既然全是小写字母,那么数组大小就确定了。
代码:
class Solution {
public boolean canConstruct(String ransomNote, String magazine) {
int [] res=new int[26];
for(int i=0;i<magazine.length();i++){
res[magazine.charAt(i)-'a']++;
}
for(int i=0;i<ransomNote.length();i++){
res[ransomNote.charAt(i)-'a']--;
}
for(int i=0;i<26;i++){
if(res[i]<0)
return false;
}
return true;
}
}
15.三数之和
给你一个整数数组 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] 。
注意,输出的顺序和三元组的顺序并不重要。
题解:双指针解法。注意去重问题(元素值不能相同)。
代码:
class Solution {
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;
//对a去重
if(i>0 && nums[i]==nums[i-1]){
continue;
}
int left=i+1,right=nums.length-1;
while(left<right){
int sum=nums[i]+nums[left]+nums[right];
if(sum<0){
left++;
}else if(sum>0){
right--;
}else {
res.add(Arrays.asList(nums[i],nums[left],nums[right]));
//找到b和c之后,对二者去重
while(right>left && nums[right]==nums[right-1]) right--;
while(right>left && nums[left]==nums[left+1]) left++;
left++;
right--;
}
}
}
return res;
}
}
18.四数之和
给你一个由 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]]
题解:在三数之和的基础上添加一个for循环。这个去重是数组下标的去重。
代码:
class Solution {
//在三数之和的基础上加一个for循环
public List<List<Integer>> fourSum(int[] nums, int target) {
List<List<Integer>> res=new ArrayList<>();
Arrays.sort(nums);
for(int j=0;j<nums.length;j++){
if(nums[j]>0 && nums[j]>target) {
return res;
}
//去重
if(j>0 && nums[j]==nums[j-1]){
continue;
}
for(int i=j+1;i<nums.length;i++){
//对a去重
if(i>j+1 && nums[i]==nums[i-1]){
continue;
}
int left=i+1,right=nums.length-1;
while(left<right){
int sum=nums[i]+nums[left]+nums[right]+nums[j];
if(sum<target){
left++;
}else if(sum>target){
right--;
}else {
res.add(Arrays.asList(nums[j],nums[i],nums[left],nums[right]));
//找到b和c之后,对二者去重
while(right>left && nums[right]==nums[right-1]) right--;
while(right>left && nums[left]==nums[left+1]) left++;
left++;
right--;
}
}
}
}
return res;
}
}