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

454.四数相加II 

力扣题目链接

 思路

因为不用但心重复元素，所以大胆使用map

代码实现

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 cnt = 0;
        for (int c: nums3) {
            for (int d: nums4) {
                if (umap.find(0 - (c + d)) != umap.end()) {
                    cnt += umap[0-(c+d)];
                }
            }
        }
        return cnt;
    }
};

收获

map[a+b]++ 这一段代码的意思是，map的一个key值是a+b，value+1。

383.赎金信

力扣题目链接

 思路

这道题和242.有效的字母异位词有曲艺同工之处。

设置一个数组rec[26]，如果一个字符在ransomNote中出现，rec[ransomNote[i]-'a'] ++。

如果在magazine中出现，rec[magazine[i]-'a'] --。

最后判断，这个数组有没有大于零的，有的话就返回false，没有就返回true。

代码实现

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        int rlen = ransomNote.size();
        int mlen = magazine.size();
        int rec[26];
        memset(rec, 0, sizeof(rec));

        for(int i = 0 ; i < rlen; i ++)
        {
            rec[ransomNote[i]-'a'] ++;
        }
        for(int i = 0 ; i < mlen; i ++)
        {
            rec[magazine[i]-'a'] --;
        }
        for(int i =0 ; i < 26; i ++)
        {
            if(rec[i]>0)
                return false;
        }
        return true;
    }
};

15.三数之和

力扣题目链接

代码实现

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

 18.四数之和

力扣题目链接

 代码实现

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