代码随想录Day7（454.四数相加II，383. 赎金信，15. 三数之和，18. 四数之和）

代码随想录Day7（454.四数相加II，383. 赎金信，15. 三数之和，18. 四数之和）

454.四数相加II

思路：

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

        for(int i : nums1){
            for(int j : nums2){
                map[i+j]++;
            }
        }

        int count = 0;

        for(int i : nums3){
            for(int j : nums4){
                if(map.find(0-(i+j)) != map.end()){
                    count += map[0-(i+j)];
                }
            }
        }

        return count;
    }
};

383. 赎金信

思路：

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {

        unordered_map<char, int> map;

        for(char item : magazine){
            map[item]++;
        }

        for(char item : ransomNote){
            if(map.find(item)!=map.end()){
                map[item]--;
                if(map[item]<0){
                    return false;
                }
            }
            else{
                return false;
            }
        }

        return true;

    }
};

15. 三数之和

思路：双指针left=i+1，right=nums.size()-1，注意去重

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {

        set<vector<int>> result;

        sort(nums.begin(), nums.end());

        for(int i=0; i<nums.size()-2; i++){

            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{
                    result.insert(vector<int>{nums[i],nums[left],nums[right]});
                    left++;
                    right--;
                }
            }          
        }

        // set去重后转回vecotr
        vector<vector<int>> result2;
        result2.assign(result.begin(), result.end());

        return result2;      
    }
};

18. 四数之和

思路：

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {

        vector<vector<int>> result2;
        if(nums.size()<4){
            return result2;
        }

        set<vector<int>> result;

        sort(nums.begin(), nums.end());


        for(int i=0; i<nums.size()-3; i++){
            // if(nums[i]>=target && nums[i]>0){
            //     break;
            // }
            for(int j=i+1; j<nums.size()-2; j++){
                // if(nums[i]+nums[j]>=target && nums[i]+nums[j]>0){
                //     break;
                // }
                int left=j+1;
                int right=nums.size()-1;
                // if(nums[right]<=target && nums[right]<0){
                //     break;
                // }
                while(left<right){
                    long sum = (long)nums[i]+nums[j]+nums[left]+nums[right];
                    if(sum<target){
                        left++;
                    }
                    else if(sum>target){
                        right--;
                    }
                    else{             
                        result.insert(vector<int>{nums[i],nums[j],nums[left],nums[right]});
                        left++;
                        right--;
                    }
                }
            }
        }
        result2.assign(result.begin(), result.end());

        return result2;

    }
};

总结

• 454.四数相加II (opens new window)是四个独立的数组，只要找到A[i] + B[j] + C[k] + D[l] = 0就可以，不用考虑有重复的四个元素相加等于0的情况。
• 15.三数之和 (opens new window)的双指针解法是一层for循环num[i]为确定值，然后循环内有left和right下标作为双指针，找到nums[i] + nums[left] + nums[right] ==0，四数之和加一层循环，五数六数以此类推。