打卡第七天：四数相加II、赎金信、三数之和、四数之和

1、四数相加II

题目

文章

视频

思路：统计A和B中元素都相互相加一次以后，出现的和的数值及其次数（++）。数值为key，次数为value，放到unordered_map中。然后再遍历C和D元素相互相加，如果0 -（c+d）出现过，则是满足条件的一组数。返回统计值。

class Solution {
public:
    int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
        unordered_map<int,int> map;
        for (int a : A){
            for (int b : B){
                map[a + b]++;//在map中记录两数组元素的和与该和出现的次数
            }
        }
        int count = 0;
        for (int c : C){
            for (int d : D){
                //如果0-（c+d）在map中出现过，则把map里key对应的value（元素和出现的次数统计进count
                if (map.find(0 - (c + d)) != map.end()){
                    count += map[0 - (c + d)];
                }
            }
        }
        return count;
    }
};

2、赎金信

题目

文章

类似字母异位词题目，用一个长度为26的数组来记录magazine里字母出现的次数。

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        int record[26] = {0};
        if (ransomNote.size() > magazine.size()){//ransomNote不应当比magazine长
            return false;
        }
        for (int i = 0; i < magazine.size(); i++){
            record[magazine[i] - 'a']++;
        }
        for (int j = 0; j < ransomNote.size(); j++){
            record[ransomNote[j]-'a']--;
            if (record[ransomNote[j]-'a'] < 0){
                return false;
            }
        }
        return true;
    }
};

3、三数之和

题目

文章

视频

思路：

哈希法难以做到bug free，去重很难考虑周全。考虑双指针法。

将数组排序后，i指向数组的第一个元素往后遍历（记为a）。在这个大的遍历框架内，left指向i+1开始向后（记为b），right指向结尾处开始向前（记为c）。需要的就是满足a+b+c=0的元素组合，且组合不能重复，组合内元素可以重复。a不变，b是控制变大的元素，c是控制变小的元素。

当a+b+c>0时，需要c变小，则right向前移动一位，继续判断a+b+c。当a+b+c<0时，需要b变大，则left向后移动一位，继续判断a+b+c。直到遍历结束，i++。left从此时的i+1继续开始，right从结尾处继续开始。

看到群里讨论，说a+b+c<0时为什么是left++而不是right++或者a+b+c>0时为什么是right--而不是left--？

首先，无论是a+b+c是大于0还是小于0，开始判断的时候只有三种情况，第一种是起始时或者同时left++和right--，这时当然left不能减且right不能加。第二种是刚操作过left++的时刻，这是已经判断过a+前一个b+c<0的结果，这种left++会一直操作到a+b+c等于或大于零。等于零那自然是满足条件，如果大于零，那么就说明，对于此时的c，没有合适的b能让他们三者的和为0，这个满足条件的b一定是大于left-1指向的b而小于left指向的b，因此在这一轮i的循环里，这个c值就被舍弃了，不可能再出现在有这轮的i存在的解集里，所以如果后面有a+b+c<0时把right++是无意义的，c值要么参与了等于0满足条件，要么就被舍弃了。那么第三种 刚操作过right--的时刻和第二种一样的道理。

除了双指针思想，另一个关键在于去重。判断语句：

nums[i] == nums[i - 1]//判断这个条件满足时，将这个组合舍弃

也就是说，第i个元素和第（i-1）个元素相等时，将这个组合舍弃。为什么呢？首先数组是有序的，如果有满足a = nums[i-1]时的元素组合，则一定会把所有可能性都加入其中。那么就不再需要以这个元素为a的元素组合了。

为什么不能是判断nums[i] == nums[i+1]？因为有可能b = nums[i] = nums[i-1]（a、b、c非0时不可能都相等，即不可能a=c），那就把这个组合给排除了，这是错误的。

比如nums = {-4, -1, -1, -1, 2, 4}有{-1, -1, 2}的组合。当a走到下标为1的-1时，用nums[i] == nums[i+1]的判断就把这个组合舍弃了，后面亦然。而用和前一个元素比较的判断时，当a走到下标为2的-1，也有{-1, -1, 2}的组合，但是是重复的，这样就把这个重复的排除了。

最后，在找到的三元组中对b和c各自去重。

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++){
            if (nums[i] > 0){//排序后若第一个元素大于0，则不可能有三元素和为0
                return result;
            }
            //注意去重方法，和前一个元素比较
            if (i > 0 && nums[i] == nums[i - 1]){//别忘了i>0
                continue;//跳过这个情形
            }
            int left = i + 1;
            int right = nums.size() - 1;
            while (right > left){
                if (nums[i] + nums[left] + nums[right] > 0) right--;
                else if (nums[i] + nums[left] + nums[right] < 0) left++;
                else{
                    result.push_back(vector<int>{nums[i], nums[left], nums[right]});//push_back()
                    //有相等的相邻b，left向后移，直到与后一个元素不相等，完成去重
                    while (right > left && nums[right] == nums[right - 1]) right--;
                    //有相等的相邻c，right向前移，直到与前一个元素不相等，完成去重
                    while (right > left && nums[left] == nums[left + 1]) left++;
                    right--;
                    left++;//对相同的i，有合适的left和right后同时移动。
                }
            }
        }
        return result;
    }
};

在对b和c去重时不要忘记条件里要有right>left，否则会出错。

4、四数之和

题目

文章

视频

思路：两层for循环nums[k] + nums[i]为确定值，依然是循环内有left和right下标作为双指针，找出nums[k] + nums[i] + nums[left] + nums[right] == target的情况，三数之和的时间复杂度是O(n^2)，四数之和的时间复杂度是O(n^3) 。

关键：注意剪枝。不能按三数之和为0那题的思维惯性，直接在剪枝时用大于target就了事。因为这样没有考虑到target是负数的情况。比如target是-4，不能因为第一个元素是-3就返回无结果。因为第二个元素也可以是负数，而负数相加是会变小的，这和三数之和为0的题有所区别。

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> result;
        sort(nums.begin(), nums.end());
        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++){//四数与三数的区别，多一层for
                if (nums[k] + nums[i] > target && nums[k] + nums[i] >= 0){//2级剪枝
                    break;
                }
                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[k] + nums[i] + nums[left] + nums[right] > target){
                        right--;
                    }else if ((long) nums[k] + nums[i] + nums[left] + nums[right] < target){
                        left++;
                    }else{
                        result.push_back(vector<int>{nums[k], 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 result;
    }
};

明天加油。