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

454.四数相加II

解法一:暴力解法,四层for循环,超时了

解法二:哈希法
思路:
1、首先定义 一个unordered_map,key 放a和b两数之和,value 放a和b两数之和出现的次数。
2、遍历nums1和nums2数组,统计两个数组元素之和,和出现的次数,放到map中。
3、定义int变量count,用来统计 a+b+c+d = 0 出现的次数。
4、再遍历nums3和nums4数组,找到如果 0-(c+d) 在map中出现过的话,就用count把map中key对应的value也就是出现次数统计出来。
5、最后返回统计值 count 就可以了

注意:
1、for(int a :nums1)是C++ 11里的用法,表示从数组nums1依次取出元素赋值给整型变量a。
2、本题是使用 map 巧妙解决的问题,好好体会一下 哈希法 如何提高程序执行效率,降低时间复杂度,当然使用 哈希法 会提高空间复杂度,但一般来说我们都是舍空间 换时间, 工业开发也是这样。

完整代码如下:

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int, int> map;//key:a+b的数值,value:a+b数值出现的次数
        // 遍历大A和大B数组,统计两个数组元素之和,和出现的次数,放到map中
        for(int a : nums1){
            for(int b : nums2){
                map[a + b]++;
            }
        }

        int count = 0;// 统计a+b+c+d = 0 出现的次数
        // 在遍历大C和大D数组,找到如果 0-(c+d) 在map中出现过的话,就把map中key对应的value也就是出现次数统计出来.
        for(int c : nums3){
            for(int d : nums4){
                if(map.find(0 - (c + d)) != map.end()){
                    count += map[0 - (c + d)];
                }
            }
        }
        return count;
    }
};

383. 赎金信

思路:
1、判断一个元素是否出现在集合里,用哈希法
2、只有26个字母,可用数组
3、先把父集(magazine)存进哈希表,再用子集(ransomNote)减去
4、减的过程中发现有负数,则说明是false,否则true

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        int hash[26] = {0};
        if(magazine.size() < ransomNote.size()){
            return false;
        }
        for(int a : magazine){
            hash[a - 'a']++;
        }
        for(int b : ransomNote){
            hash[b - 'a']--;
            // 如果小于零说明ransomNote里出现的字符,magazine没有
            if(hash[b - 'a'] < 0){
                return false;
            }
        }
        return true;

    }
};

15. 三数之和

采用双指针解法
注意:
1、先对集合进行排序
2、对i去重时,要使i > 0,防止i - 1为负数
3、对i去重时,是nums[i] == nums[i - 1],而不是nums[i] == nums[i + 1]。若集合为{-1,-1,2},采用后一种写法则会导致这个结果被跳过,因此采用前一种写法。
nums[n-1]是对结果集的个数去重,nums[n]==nums[n+1]是对结果集的内容去重
4、先收集结果,再对leftright去重
5、不需要去重的直接移动收缩双指针即可

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++){
            //排序后第一个元素>0,必不可能三数之和为0
            if(nums[i] > 0) return result;
            //对i进行去重,三元组不可重复,直接continue
            if(i > 0 && nums[i] == nums[i - 1]) continue;
            int left = i + 1;
            int right = nums.size() - 1;
            //开始移动双指针
            while(left < right){
                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]});
                    //对left和right进行去重
                    while(left < right && nums[left] == nums[left + 1]) left++;
                    while(left < right && nums[right] == nums[right - 1]) right--;

                    //收集结果后收缩双指针
                    right--;
                    left++;
                }
            }
        }
        return result;
    }
};

18. 四数之和

采用双指针解法
注意: 不强制转化为long的话会溢出,因为题目中target的值可以到达10^9

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 && target > 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 && target > 0){
                    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[i] + nums[k] + nums[left] + nums[right] < target){
                        left++;
                    }else{
                        result.push_back(vector<int>{nums[k], nums[i], nums[left], nums[right]});
                        //对left和right进行去重
                        while(right > left && nums[left] == nums[left + 1]) left++;
                        while(right > left && nums[right] == nums[right - 1]) right--;
                        //收集结果后收缩双指针
                        left++;
                        right--;
                    }
                }
            }
        }
        return result;
    }
};
  • 5
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值