【代码随想录第7天|454.四数相加II ,383. 赎金信 ,15. 三数之和,18. 四数之和】

454.四数相加II

思路

利用unordered_map<int, int>进行键值对的记录，四个数组两两一组，两组的和互为相反数则存在。

代码随想录

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

383. 赎金信

思路

赎金信写成的条件是，信里的字母magazine里字母的子集，因此先用数组的键值对确认magazine中每个字母的键值对，然后作减法判断字母是否足够。

代码随想录

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

15. 三数之和

思路

双指针+for循环即可实现类似三重遍历的效果，sort()的目的是为了避免重复的输出

代码随想录

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> result;
        sort(nums.begin(), nums.end());
        for (int i = 0; i < nums.size(); i++) {
            int left = i + 1;
            int right = nums.size() - 1;
            // 判断nums[i]是否在之前已经判断过
            if (i > 0 && nums[i] == nums[i-1]) {continue;}
            while (left < right) {
                if (nums[i] + nums[left] + nums[right] == 0) {
                    result.push_back(vector<int>{nums[i], nums[left], nums[right]});
                    // 判断 right 是否重复
                    while (right > left && nums[right] == nums[right-1]) right--;
                    // 判断 left 是否重复
                    while (right > left && nums[left] == nums[left+1]) left++;
                    right--;
                    left++;
                }
                else if (nums[i] + nums[left] + nums[right] < 0) {
                    left ++;
                }
                else {right--;}
            }
        }
        return result;
    }
};

收获

需要进行三个关键判断（剪枝）:

1. 判断 nums[i] 是否在之前已经判断过
2. 判断 right 是否重复
3. 判断 left 是否重复

18. 四数之和

思路

每一次剪枝的时候都要考虑和是否大于0，因为对于负值进行剪枝在某些case是会出错误的；
在三数之和的基础上多一层for循环。

代码随想录

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        sort(nums.begin(), nums.end());
        vector<vector<int>> result;
        for (int k = 0; k < nums.size(); k++) {
            // 第一次剪枝
            if (nums[k] > target && nums[k] >= 0) {
                break;
            }
            if (k > 0 && nums[k] == nums[k -1]) {
                continue;
            }
            for (int i = k + 1; i < nums.size(); i++) {
                // 第二次剪枝
                if (nums[i] + nums[k] > target && nums[i] + nums[k] >= 0) {
                    break;
                }
                // 对 nums[i] 去重
                if (i > k + 1 && nums[i] == nums[i-1]) {
                    continue;
                }
                int left = i + 1;
                int right = nums.size() - 1;
                while (right > left) {
                    if ( (long) nums[i] + nums[k] + nums[left] + nums[right] > target) {
                        right--;
                    } else if ((long) nums[i] + nums[k] + nums[left] + nums[right] < target) {
                        left++;
                    } else {
                        result.push_back(vector<int> {nums[i], nums[k], 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 result;
    }
};

收获

双指针指示两个值，因此还需要两个循环来指示剩下的两个值。