代码随想录算法训练营Day6

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

454.四数相加II 题目链接

本题解题步骤：

首先定义 一个Hashmap，key放a和b两数之和，value 放a和b两数之和出现的次数。

遍历A和B数组，统计两个数组元素之和key，和出现的次数value，放到map中。

定义int变量count，用来统计 a+b+c+d = 0 出现的次数。

在遍历C和D数组，如果找到 0-(c+d) 在map中出现过的话，就用count把map中key对应的value也就是出现次数统计出来。

最后返回统计值 count 就可以了

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        int count = 0;
        Map<Integer, Integer> map = new HashMap<>();
        for (int i : nums1){
            for (int j : nums2){
                //如果map中key存在i +j，则+1， 如果不存在则赋值1
                 map.put(i + j, map.getOrDefault(i + j , 0)+ 1);
            }
        }
        for (int i : nums3){
            for (int j : nums4){
                if(map.containsKey(0 - (i + j))){
                    count += map.get(0 - (i + j));
                }
            }
        }
        return count;
    }
}

时间复杂度: O(n)

---

383. 赎金信 题目链接

用一个长度为26的数组还记录magazine里字母出现的次数,然后再用ransomNote去验证这个数组是否包含了ransomNote所需要的所有字母。

class Solution {
    public boolean canConstruct(String ransomNote, String magazine) {
        if (ransomNote.length() > magazine.length()) return false;
        int [] res = new int[26];
        for (char c : magazine.toCharArray(){
            res[c - 'a']++;
        }
        for (char c : ransomNote.toCharArray(){
            res[c - 'a']--;
            if(res[c - 'a'] < 0){
                 return false;
            }       
        }
        return true;
            
    }                                                                                                                            
}

• 时间复杂度: O(n)
• 空间复杂度: O(1)

  ---

  15. 三数之和 题目链接

  在这里采用双指针法，思路更为直观。需要注意的是去重逻辑与操作。

• class Solution {
      public List<List<Integer>> threeSum(int[] nums) {
          List<List<Integer>> res = new ArrayList<>();
          int len = nums.length;
          Arrays.sort(nums);                        //排序
          if (nums == null || len < 3) return res;
          for (int i = 0 ; i < len; i++){
              if (nums[i] > 0)break;             //第一个数都>0 后面全是,不成立
              if (i > 0 && nums[i] == nums[i-1]) continue;  //去重
              
              int l = i + 1, r = len - 1;
              while(l < r){
                  int sum = nums[i] + nums[l] + nums[r];
                  if (sum == 0){
                      res.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 res;
      }
  }

  ---

  18. 四数之和 题目链接

  与三数之和类似，采用双指针法，注意的是去重与剪枝操作

  class Solution {
      public List<List<Integer>> fourSum(int[] nums, int target) {
          List<List<Integer>> res = new ArrayList<>();
          int len = nums.length;
          Arrays.sort(nums);                        //排序
          if (nums == null || len < 4)return res;
  
          for (int i = 0 ; i < len; i++){
              if (nums[i] > 0 && nums[i] > target)break;    //剪枝
              if (i > 0 && nums[i] == nums[i -1])continue;  //去掉重复
  
              for(int j = i + 1; j < len; j++){
                  if (j > i + 1 && nums[j] == nums[j - 1] )continue;  //去重
                  int l = j +1 , r = len - 1;
                  while(l < r){
                      int sum = nums[i] + nums[j] +nums[l] + nums[r];  
                      if(sum < target)l++;            
                      else if(sum > target)r--;
                      else{
                          res.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--;
                      }
                  }
              }
          }
          return res;
      }
  }