代码随想录第七天

四数相加II

454. 四数相加 II - 力扣（Leetcode）

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int,int> map;//key:a+b的数值，val：a+b的次数
        int ans=0;
        //遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中
        for(int i=0;i<nums1.size();i++){
            for(int j=0;j<nums2.size();j++){
                auto iter=map.find(nums1[i]+nums2[j]);
                if(iter!=map.end()){
                    iter->second++;
                }
                map.insert(pair<int,int>{nums1[i]+nums2[j],1});
            }
        }
        for(int k=0;k<nums3.size();k++){
            for(int l=0;l<nums4.size();l++){
                auto iter=map.find(-(nums3[k]+nums4[l]));
                if(iter!=map.end()){
                    ans+=(iter->second);
                }
            }
        }
        return ans;
    }
};

赎金信

[383. 赎金信 - 力扣（Leetcode）

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        unordered_map<char,int>map;
        for(int i=0;i<magazine.size();i++){
            auto iter=map.find(magazine[i]);
            if(iter!=map.end()){
                iter->second++;
            }
            map.insert(pair<char,int>{magazine[i],1});
        }
        for(int j=0;j<ransomNote.size();j++){
            auto iter=map.find(ransomNote[j]);
            if(iter!=map.end()&&(iter->second)>0){
                (iter->second)--;
            }
            else
            return false;
        }
        return true;
    }
};

三数之和

15. 三数之和 - 力扣（Leetcode）

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ans;
        sort(nums.begin(),nums.end());
        for(int i=0;i<nums.size();i++){
            if(nums[i]>0) return ans;
            if(i>0&&nums[i]==nums[i-1]){
                continue;
            }
            int left=i+1,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{
                    ans.push_back(vector<int>{nums[i],nums[left],nums[right]});
                    while(left<right&&nums[left]==nums[left+1]){
                        left++;
                    }
                    while(left<right&&nums[right]==nums[right-1]){
                        right--;
                    }
                    left++;
                    right--;
                }
            }
        }
        return ans;
    }
};

a去重操作

if (i > 0 && nums[i] == nums[i - 1]) {
    continue;
}

 这么写就是当前使用 nums[i]，我们判断前一位是不是一样的元素，在看 {-1, -1 ,2} 这组数据，当遍历到 第一个 -1 的时候，只要前一位没有-1，那么 {-1, -1 ,2} 这组数据一样可以收录到 结果集里。

b与c去重操作

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

四数之和

18. 四数之和 - 力扣（Leetcode）

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