代码随想录 - 训练营8期- day 7

● 454.四数相加II
● 383. 赎金信
● 15. 三数之和
● 18. 四数之和
● 总结

454. 四数相加 II

统计个数，难度还行

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        int cnt = 0;
        unordered_map<int, int> hash;
        for(int i = 0; i < nums1.size(); ++i) {
            for(int j = 0; j < nums2.size(); ++j) {
                hash[ nums1[i] + nums2[j] ]++;
            }
        }
        for(int i = 0; i < nums3.size(); ++i) {
            for(int j = 0; j < nums4.size(); ++j) {
                if(hash.find( 0 - (nums3[i] + nums4[j]) ) != hash.end()) {
                    cnt += hash[ 0 - (nums3[i] + nums4[j]) ];
                }
            }
        }
        return cnt;
    }
};

383. 赎金信

判断条件改改，和昨天那个一样的

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        int letter[26] = {};
        for(char& c : ransomNote) {
            letter[c - 'a'] ++;
        }
        for(char& c : magazine) {
            letter[c - 'a'] --;
        }
        for(int i = 0; i < 26; ++i) {
            if(letter[i] > 0) {
                return false;
            }
        }
        return true;
    }
};

15. 三数之和

梦破碎的地方

学到了，注意去重。

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

18. 四数之和

一次过了，非常开心

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        using __int64 = long long;
        sort(nums.begin(), nums.end());
        vector<vector<int>> res;
        int n = nums.size(); 
        for(int i = 0; i < n - 3; ++i) {
            if(i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }
            for(int j = i + 1; j < n - 2; ++j) {
                if(j > (i + 1) && nums[j] == nums[j - 1]) {
                    continue;
                }
                int l = j + 1;
                int r = n - 1;
                __int64 num = nums[i] + nums[j];
                while(l < r) {
                    if( __int64(num + nums[l] + nums[r]) > target) r--;
                    else if( __int64(num + nums[l] + nums[r]) < target) l++;
                    else {
                        res.push_back( {nums[i], nums[j], nums[l], nums[r]} );
                        l++;
                        r--;
                        while(l < r && nums[l] == nums[l - 1]) l ++;
                        while(l < r && nums[r] == nums[r + 1]) r --;
                    }
                }
            }
        }
        return res;
    }
};