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

454.四数相加II

由于此题不用考虑有重复的四个元素相加等于0的情况，所以用哈希表最合适。

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

383. 赎金信

使用数组：

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        int record[26] = {0};
        //add
        if (ransomNote.size() > magazine.size()) {
            return false;
        }
        for (int i = 0; i < magazine.length(); i++) {
            // 通过recode数据记录 magazine里各个字符出现次数
            record[magazine[i]-'a'] ++;
        }
        for (int j = 0; j < ransomNote.length(); j++) {
            // 遍历ransomNote，在record里对应的字符个数做--操作
            record[ransomNote[j]-'a']--;
            // 如果小于零说明ransomNote里出现的字符，magazine没有
            if(record[ransomNote[j]-'a'] < 0) {
                return false;
            }
        }
        return true;
    }
};

使用哈希表：

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        unordered_map<int,int> umap;
        for(auto m: magazine){
            umap[m]++;
        }
        for(auto n: ransomNote){
            if(--umap[n] <0) return false;
        }
        return true;
    }
};

此处使用数组执行时间更快，原因在于如果使用哈希表，需要维持哈希表，还要计算哈希函数。因为此处a-z突出一个有序，所以用数组更加快捷。若是保存无序的，那么只能使用哈希表。

15. 三数之和

双指针法：

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        vector<vector<int>> res;
        
        for(int i = 0;i<nums.size();i++){
            //如果排序之后，第一个数大于零，那么之后的肯定都大于零。直接结束
            if(nums[i] > 0) return res;
            //排除重复的三元组
            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){
                    left++;
                }else if(nums[i] + nums[left]+nums[right] > 0){
                    right--;
                }else{
                    res.push_back(vector<int>{nums[i],nums[left],nums[right]});
                    //如下 left < right 条件不可省略，因为 left和right的值已经改动
                    //并且如下操作也是为了排除重复的三元组
                    while(left < right&&nums[right] == nums[right-1]) right--;
                    while(left < right&&nums[left] == nums[left+1]) left++;
                    right--;left++;
                }
            }   
        }    
        return res;    
    }
};

18. 四数之和

和15题有一曲同工之妙，双指针法：

只不过需要进行四个数的加法运算，所以在外面再嵌套一个for循环。另外，由于此题的target不再固定为0，而是一个变换的数，需要注意判定的条件。

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> res;
        sort(nums.begin(),nums.end());
        for(int k = 0;k<nums.size();k++){
            if(nums[k]>target && nums[k]>0) break;
            if(k>0 && nums[k] == nums[k-1]){
                continue;
            }
            for(int i = k+1;i<nums.size();i++){
               if(nums[i]+nums[k]>target && nums[k]+nums[i]>0) break;
               if(i>k+1 && nums[i] == nums[i-1]){
                    continue;
                } 
                int left = i + 1;
                int right = nums.size() - 1;
                while (right > left) {
                    // nums[k] + nums[i] + nums[left] + nums[right] > target 会溢出
                    if ((long) nums[k] + nums[i] + nums[left] + nums[right] > target) {
                        right--;
                    // nums[k] + nums[i] + nums[left] + nums[right] < target 会溢出
                    } else if ((long) nums[k] + nums[i] + nums[left] + nums[right]  < target) {
                        left++;
                    } else {
                        res.push_back(vector<int>{nums[k], nums[i], nums[left], nums[right]});
                        // 对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;
    }
};