[代码随想录算法训练营]刷题笔记 Day 7

第454题.四数相加II

将四组数组的数值两两配对，其中一组储存在map里，另一组用来遍历map是否出现过对应的和

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        // 分成两个两个一组做遍历
        unordered_map<int, int> map;
        int count = 0;

        for(auto a : nums1){
            for(auto b : nums2){
                map[a+b]++; // 若没有找到对应的键值会自动新增，初始化为0
            }            
        }

        for(auto c : nums3){
            for(auto d : nums4){
                int target = 0 - (c + d);
                if(map.find(target) != map.end()){
                    count += map[target];
                }
            }
        }

        return count;
    }
};

383. 赎金信

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        int table[26] = {0};

        for(auto c : magazine){
            table[c - 'a']++;
        }

        for(auto c : ransomNote){
            table[c - 'a']--;
            if(table[c - 'a'] < 0) return false;
        }

        return true;
    }
};

第15题. 三数之和

这题运用双指针比哈希表更好做，因未要判断是否重复。
思路是运用一个回圈，然后双指针指向左右值来判定和是否等于0。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> result;
        sort(nums.begin(), nums.end());

        // a + b + c = 0
        for(int i = 0; i < nums.size(); i++){
            // 如果a>0，因为数组已经经过排序，后面的数值必大于0
            // 其三树枝和不可能为0，故跳过
            if(nums[i] > 0) break;

            // 对a进行查重，往前查是因为不要干扰b和c
            if(i > 0 && nums[i] == nums[i - 1]) continue; 

            // 选定了a，利用双指针来找b和c
            int left = i + 1;
            int right = nums.size() - 1;
            // while条件判断为(left > right)，如果等于也在判定式的话会出现left = right然后三数之和也等于0的情况，这样实际上只有两数之和
            while(right > left){
                if(nums[i] + nums[left] + nums[right] > 0)
                    right--;
                else if(nums[i] + nums[left] + nums[right] < 0)
                    left++;
                else{
                    // 因为保证了a，b，c不会重复 所以我们不必去判断答案是否有重复
                    result.push_back(vector<int>{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++;

                    // 找到答案时，双指针同时收缩
                    right--;
                    left++;
                }
            }
        }

        return result;
    }
};

第18题. 四数之和

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> result;
        // std::sort()
        // template <class RandomAccessIterator>  void sort (RandomAccessIterator first, RandomAccessIterator last);
        sort(nums.begin(), nums.end());

        // a + b + c + d = target
        // 选定a的值
        for(int i = 0; i < nums.size(); i++){

            // *--------------------------剪枝-----------------------------*
            // 如果target是一个正数，而遍历的第一个值又大于他。
            // 因为我们已经将nums排序过，后面的2、3、4个值只会越来越大，肯定不符合条件。
            // 而往后面所选的值肯定越来越大，都不符合等式，所以我们直接结束遍历
            if(target > 0 && nums[i] > target)
                break;

            // *--------------------------查重-----------------------------*
            // 若a值等于前面一个值跳过这圈
            if(i > 0 && nums[i] == nums[i-1])
                continue;

            //选定b的值
            for(int j = i + 1; j < nums.size(); j++){
                
                // *--------------------------剪枝-----------------------------*
                // a + b 如果已经大于target，加上c和d肯定也会大于target
                if(target > 0 && (nums[i] + nums[j]) > target)
                    break;

                // *--------------------------查重-----------------------------*
                // 若b值等于前面一个值跳过这圈
                if(j > i + 1 && nums[j] == nums[j-1])
                    continue;


                // 双指针找c和d
                int left = j + 1;
                int right = nums.size() - 1;

                // 避免c和d选到同一个值
                while(right > left){

                    // 判断a + b + c + d的值
                    if((long)nums[i] + nums[j] + nums[left] + nums[right] > target)
                        right--;
                    else if((long)nums[i] + nums[j] + nums[left] + nums[right] < target)
                        left++;
                    else{
                        result.push_back(vector<int>{nums[i], nums[j], 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 result;
    }
};