代码随想录算法训练营day7-哈希表part2

四数相加 II

原题链接：四数相加 II

思路：将四个数组分为两组，遍历第一组，将其和使用哈希表记录下来，由于可能重复，故需要统计和的数量。遍历另外一组，根据其和的相反数去查询哈希表，如果存在，则使结果加上其出现次数

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {

        int res=0;
        unordered_map<int,int>m;
        for(auto i : nums1){
            for(auto j : nums2){
                m[i+j]++;
            }
        }

        for(auto i : nums3){
            for(auto j : nums4){
                int t = -(i+j);
                if(m.find(t)!=m.end()){
                    res+=m[t];
                }
            }
        }

        return res;
    }
};

 赎金信

原题链接：赎金信

思路：题目可以转化为，某个数出现的次数。判断是否出现，以及出现次数，容易想到使用哈希表。由于字母数量较少，将magazine存入数组中，而后遍历另一数组，如果某个字母出现负数，则失败。

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

        for(auto i : magazine){
            nums[i-'a']++;
        }

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

        return true;
    }
};

 三数之和

原题链接：三数之和

思路：容易想到，遍历两遍，然后判断差值是否存在，但是这样做，去重十分麻烦；可以使用双指针，先将数组排序，排序后有个特征，越靠后的数字越大，利用这点，可以使用两个变量从两端往中间逼近来获取符合的值。

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()-2;i++){
            if(i>0&&nums[i]==nums[i-1]){
                continue;
            }
            if(nums[i] > 0){
                break;
            }
            int left = i+1;
            int right = nums.size() - 1;
            while(left<right){
                int sum = nums[i]+nums[left]+nums[right];
                if(sum < 0){
                    left++;
                }else if(sum > 0){
                    right--;
                }
                else{
                    res.push_back({nums[i],nums[left],nums[right]});
                    left++;
                    right--;
                    while(left<right&&nums[left] == nums[left-1]) left++;
                    while(left<right&&nums[right] == nums[right+1]) right--;
                }
            }
        }

        return res;
    }
};

 四数之和

原题链接：四数之和

思路：思路和三数之和类似，只是要枚举两个变量。

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        //
        if(nums.size() < 4){
            return {};
        }
        sort(nums.begin(),nums.end());

        vector<vector<int>>res;

        for(int i = 0;i<nums.size() - 3;i++){
            if(i>0&&nums[i]==nums[i-1]){
                continue;
            }
            for(int j = i+1;j<nums.size()-2;j++){
                if(j>i+1&&nums[j]==nums[j-1]){
                    continue;
                }
                int left = j+1;
                int right = nums.size() - 1;
                while(right > left){
                    long long sum1 = nums[i]+nums[j];
                    long long sum2 = nums[left]+nums[right];
                    long long  sum = sum1+sum2;
                    if(sum > target) right--;
                    else if(sum < target) left++;
                    else{
                        res.push_back({nums[i],nums[j],nums[left],nums[right]});
                        left++;
                        right--;
                        while(left<right&&nums[left]==nums[left-1]) left++;
                        while(left<right&&nums[right] == nums[right+1]) right--;
                    } 
                }
            }
        }

        return res;
    }
};