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

---

前言

代码随想录打卡第七天

---

一、四数相加II

解题思路：本意通过将数组转化为map进行查找，但这样就有了不确定性，还是相当于三层遍历，时间复杂度O(n^3)，通过题解思路，将四个分成两组，将前两个之和放入map，key记录和的值，value记录该值出现的次数。

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int,int> map;
        for(int a : nums1){
            for(int b : nums2){
                map[a + b]++;
            }
        }
        int count = 0;
        for(int c : nums3){
            for(int d : nums4){
                if(map.find(0 - (c + d)) != map.end()){
                    count += map[0 - (c + d)];
                }
            }
        }
        return count;

    }
};

二、 赎金信

之前做过，但是没判断ransomNote.size() > magazine.size()的情况，该情况直接返回false。

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        int record[26] = {0};
        for(int i = 0; i < ransomNote.size(); i++){
            record[ransomNote[i] - 'a']++;
        }
        for(int i = 0; i < magazine.size(); i++){
            record[magazine[i] - 'a']--;
        }
        for(int i = 0; i < 26; i++){
            if(record[i] > 0){
                return false;
            }
        }
        return true;
    }
};

三、三数之和

解题思路：排序后，确定第一个数，然后利用双指针对后续数据从第一个数后一个和最后一个数向中间遍历，三数之和大于0，右侧指针左移，小于0，左侧指针右移。同时要注意重复元素，导致产生重复组的问题。去重应将后一个去除，利用{-1，-1，2}，如果利用去重，在第一个-1就去掉，则会丢失该元组，如果到第二个-1再去重，该元组已经加入结果。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> result;
        sort(nums.begin(),nums.end());
        for(int i = 0;i < nums.size() && nums[i] <= 0;i++){
            //去重
            if(i > 0 && nums[i] == nums[i - 1]){
                continue;
            }
            int left = i + 1;
            int right = nums.size() - 1;
            while(left < right){

                if(nums[i] + nums[left] + nums[right] > 0){
                    right--;
                }else if(nums[i] + nums[left] + nums[right] < 0){
                    left++;
                }else{
                    result.push_back(vector<int>{nums[i], nums[left] , nums[right]});
                    //去重
                    while(left < right && nums[right] == nums[right - 1]) right--;
                    while(left < right && nums[left] == nums[left + 1]) left++;
                    right--;
                    left++;
                    
                }
            }
        }

        return result;
    }
};

四、四数之和

不要判断nums[k] > target 就返回了，三数之和 可以通过 nums[i] > 0 就返回了，因为 0 已经是确定的数了，四数之和这道题目 target是任意值。比如：数组是[-4, -3, -2, -1]，target是-10，不能因为-4 > -10而跳过。但是我们依旧可以去做剪枝，逻辑变成nums[i] > target && (nums[i] >=0 || target >= 0)就可以了。
四数之和为了预防求和越界，应再求和前加（long）。

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> result;
        sort(nums.begin(),nums.end());
        for(int i = 0; i < nums.size(); i++){
            //剪枝
            if(nums[i] > target && nums[i] >= 0){
                break;
            }
            //去重
            if(i > 0 && nums[i] == nums[i - 1]){
                continue;
            }
            for(int j = i + 1; j < nums.size(); j++){
                //剪枝
                if(nums[i] + nums[j] > target && nums[i] + nums[j] >= 0){
                    break;
                }
                //去重
                if(j > i + 1 && nums[j] == nums[j - 1]){
                    continue;
                }
                int left = j + 1;
                int right = nums.size() - 1;
                //printf("%d %d %d %d",i,j,left,right);
                while(left < right){
                    if((long)nums[i] + nums[j] + nums[left] + nums[right] < target){
                        left++;
                    }else if((long)nums[i] + nums[j] + nums[left] + nums[right] > target){
                        right--;
                    }else{
                        result.push_back(vector<int>{nums[i], nums[j], nums[left], nums[right]});
                        while(left < right && nums[right - 1] == nums[right]) right--;
                        while(left < right && nums[left + 1] == nums[left]) left++;
                        right--;
                        left++;
                    }
                }
            }
        }
        return result;
    }
};

五、哈希表总结

用数组不如都用map不就完事了？
用map确实可以，但使用map的空间消耗要比数组大一些，因为map要维护红黑树或者符号表，而且还要做哈希函数的运算。所以数组更加简单直接有效。
数组主要应对规模已知，且数量较小，不分散的题目；
set应用于仅需要查找单一数值题目；
map应用于需要通过值查找对应key的题目；
set和map都有set/map、multiset/map、unordered_set/map三种，前两种是基于红黑树的键值有序，最后一种无序。
代码随想录哈希表讲解

---

总结

耗时1.5小时，思路参考代码随想录提供。