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

第454题.四数相加II

题目链接

这道题目就是跟着代码随想录上面的思路来的，卡哥给的思路还是很清晰的。

主要就是把四个数组分成两组，巧妙的把问题化解，由原本的四数相加变成两数相加，然后把所有的可能加在一起返回。

本题解题步骤：

1. 首先定义 一个unordered_map，key放a和b两数之和，value 放a和b两数之和出现的次数。
2. 遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中。
3. 定义int变量count，用来统计 a+b+c+d = 0 出现的次数。
4. 在遍历大C和大D数组，找到如果 0-(c+d) 在map中出现过的话，就用count把map中key对应的value也就是出现次数统计出来。
5. 最后返回统计值 count 就可以了

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int,int> map;
        for(int a:nums1){           //遍历1,2数组，把所有a+b的情况记录在map里，有重复则+1；
            for(int b:nums2){
                map[a+b]++;
            }
        }
        int count=0;                //count用来存放和为0的次数
        for(int c:nums3){           //遍历3,4数组，找到所有a+b+c+d=0的情况，如果符合那么count加上符合的a+b的次数
            for(int d:nums4){
                if(map.find(0-(c+d))!=map.end()) count+=map[0-(c+d)];
            }
        }
        return count;
    }
};

383. 赎金信

题目链接

这道题是用数组做哈希表，26个小写字母从零开始映射到0~26的数组下标，对应数组元素存储字母出现的次数。先把magazine的每个字母存储到record中，再从record中减去ransomNote中出现的对应字母的个数，如果record出现负数说明magazine中的字母不够组成ransomNote。

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        int record[26]={0};
        if(ransomNote.size()>magazine.size()) return false;
        for(int i=0;i<magazine.size();i++){
            record[magazine[i]-'a']++;              //将magazine中所有元素存储到record数组中并记录个数
        }
        for(int i=0;i<ransomNote.size();i++){       
            record[ransomNote[i]-'a']--;            //ransomNote中出现的元素record-1，
            if(record[ransomNote[i]-'a']<0) return false;     //如果record中出现负数则说明magazine中没有出现组成ransomNote的元素或元素个数不够组成ransomNote，返回false
        }
        return true;
    }
};

第15题. 三数之和

题目链接

这道题目哈希表法复杂的去重看得我头晕眼花，本就效率不高，看看就行了，加深一下对哈希表的理解即可。双指针法要比哈希法高效一些。

解题步骤：

1. 将nums数组排序。
2. 进入循环，固定i
3. 对i位置去重
4. 将i+1设为left，nums.size()-1设为right；
5. 如果i left right对应元素之和>0，right向左移，和<0时left向右移，==0时将对应的元素写到result中，对left和right位置的元素分别去重，然后left和right指针都移动。

动画效果如下：

                             

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) return result;
            if(i>0&&nums[i]==nums[i-1]) continue;       // i去重
            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]});
                    while(right>left&&nums[right]==nums[right-1]) right--;      //right去重
                    while(right>left&&nums[left]==nums[left+1]) left++;         //left去重
                    left++;
                    right--;
                }
            }
        }
        return result;
    }
};

关于去重的思考，我觉得卡哥说的比较好，放在下面以便后来回顾的时候看起来方便：

去重逻辑的思考

a的去重

说到去重，其实主要考虑三个数的去重。 a, b ,c, 对应的就是 nums[i]，nums[left]，nums[right]

a 如果重复了怎么办，a是nums里遍历的元素，那么应该直接跳过去。

但这里有一个问题，是判断 nums[i] 与 nums[i + 1]是否相同，还是判断 nums[i] 与 nums[i-1] 是否相同。

有同学可能想，这不都一样吗。

其实不一样！

都是和 nums[i]进行比较，是比较它的前一个，还是比较它的后一个。

如果我们的写法是 这样：

if (nums[i] == nums[i + 1]) { // 去重操作
    continue;
}

那我们就把 三元组中出现重复元素的情况直接pass掉了。 例如{-1, -1 ,2} 这组数据，当遍历到第一个-1 的时候，判断 下一个也是-1，那这组数据就pass了。

我们要做的是 不能有重复的三元组，但三元组内的元素是可以重复的！

所以这里是有两个重复的维度。

那么应该这么写：

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

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

这是一个非常细节的思考过程。

b与c的去重

很多同学写本题的时候，去重的逻辑多加了 对right 和left 的去重：（代码中注释部分）

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 {
    }
}

但细想一下，这种去重其实对提升程序运行效率是没有帮助的。

拿right去重为例，即使不加这个去重逻辑，依然根据 while (right > left) 和 if (nums[i] + nums[left] + nums[right] > 0) 去完成right-- 的操作。

多加了 while (left < right && nums[right] == nums[right + 1]) right--; 这一行代码，其实就是把 需要执行的逻辑提前执行了，但并没有减少 判断的逻辑。

最直白的思考过程，就是right还是一个数一个数的减下去的，所以在哪里减的都是一样的。

所以这种去重 是可以不加的。 仅仅是 把去重的逻辑提前了而已。

 第18题. 四数之和

题目链接

相当于三数之和的加强版，多了一个数并且不再是何为0，而是给定的target；整体思路延续三数之和，使用双指针，由于加了一个数，所以加了一层for循环用来固定第二个数。

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> result;
        sort(nums.begin(),nums.end());
        for(int i=0;i<nums.size();i++){
            if(nums[i]>target&&nums[i]>=0) break;       //剪枝处理
            if(i>0&&nums[i]==nums[i-1]) continue;       //i去重
            for(int j=i+1;j<nums.size();j++){
                if(nums[i]+nums[j]>target&&nums[i]>=0) break;       //二级剪枝
                if(j>i+1&&nums[j]==nums[j-1]) continue;
                int left=j+1;
                int right=nums.size()-1;
                while(right>left){
                    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++;    //转换成long类型，否则数太大时可能导致整数溢出(Integer Overflow)
                    else{
                        result.push_back(vector<int>{nums[i],nums[j],nums[left],nums[right]});
                        while(right>left&&nums[right]==nums[right-1]) right--;           //right去重
                        while(right>left&&nums[left]==nums[left+1]) left++;              //left去重
                        right--;
                        left++;
                    }
                }
            }
        }
        return result;
    }
};

但是有一些细节需要注意，例如： 不要判断nums[k] > target 就返回了，三数之和 可以通过 nums[i] > 0 就返回了，因为 0 已经是确定的数了，四数之和这道题目 target是任意值。比如：数组是[-4, -3, -2, -1]，target是-10，不能因为-4 > -10而跳过。但是我们依旧可以去做剪枝，逻辑变成nums[i] > target && (nums[i] >=0 || target >= 0)就可以了。