算法训练Day07|哈希表part02(LeetCode454.四数相加II、LeetCode383. 赎金信、LeetCode15. 三数之和、LeetCode18. 四数之和)

3分16秒

已于 2023-08-04 17:24:43 修改

阅读量223

点赞数

分类专栏：算法与数据结构文章标签：算法散列表数据结构

于 2023-07-31 11:01:46 首次发布

本文链接：https://blog.csdn.net/weixin_43399263/article/details/131999190

版权

算法与数据结构专栏收录该内容

42 篇文章 0 订阅

订阅专栏

文章探讨了四数相加II、赎金信、三数之和和四数之和这四个编程问题的解决方案，强调了使用哈希表和双指针法来优化时间复杂度，特别是如何利用这些技巧在给定限制下找到满足条件的组合数。对于四数相加问题，提出了一种将问题简化为两部分并利用哈希表存储中间结果的方法，从而达到O(n^2)的时间复杂度。而对于三数之和的问题，通过排序和双指针技术避免了重复组合的出现。

摘要由CSDN通过智能技术生成

文章目录

文章讲解
 视频讲解

454.四数相加II

题目链接
给定四个包含整数的数组列表 A , B , C , D ,计算有多少个元组 (i, j, k, l) ，使得 A[i] + B[j] + C[k] + D[l] = 0。
为了使问题简单化，所有的 A, B, C, D 具有相同的长度 N，且 0 ≤ N ≤ 500 。所有整数的范围在 -2^28 到 2^28 - 1 之间，最终结果不会超过 2^31 - 1 。

例如:
输入:
A = [ 1, 2]
B = [-2,-1]
C = [-1, 2]
D = [ 0, 2]
输出:
2

解释:
两个元组如下:
(0, 0, 0, 1) -> A[0] + B[0] + C[0] + D[1] = 1 + (-2) + (-1) + 2 = 0
(1, 1, 0, 0) -> A[1] + B[1] + C[0] + D[0] = 2 + (-1) + (-1) + 0 = 0

重点

本题要计算的是有多少组，直接暴力解法，时间复杂度O(n^4)。

为什么用map？
有效字母异位词，只有二十六个字母，是可控的，所以用了数组。但本题数字范围不可控，而且不仅要记录数值，还要统计出现过的次数，所以用map类型。
本题解题步骤：

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

代码

之前有想过统计一个countab，统计一个countcd，最后元组数就是countab*countcd，这样也行，但是这样就要AB、CD遍历统计完之后，再遍历一遍countcd，更复杂了。

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int,int> map;
        for(int i=0;i<nums1.size();i++){
            for(int j=0;j<nums2.size();j++){
                if(map.find(nums1[i]+nums2[j])==map.end()){
                    map.insert(pair<int,int>(nums1[i]+nums2[j],1));
                }
                else{
                    map[nums1[i]+nums2[j]]++;
                }
            }
        }
        int count=0;
        for(int i=0;i<nums3.size();i++){
            for(int j=0;j<nums4.size();j++){
                if(map.find(0-(nums3[i]+nums4[j]))!=map.end()){
                    //不能是count++;
                    count+=map[0-(nums3[i]+nums4[j])];
                }
            }
        }
        return count;
    }
};

写完后再看官方答案，真的简洁了不是一点半点，for (int a : A)这种写法：

class Solution {
public:
    int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
        unordered_map<int, int> umap; //key:a+b的数值，value:a+b数值出现的次数
        // 遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中
        for (int a : A) {
            for (int b : B) {
                umap[a + b]++;
            }
        }
        int count = 0; // 统计a+b+c+d = 0 出现的次数
        // 在遍历大C和大D数组，找到如果 0-(c+d) 在map中出现过的话，就把map中key对应的value也就是出现次数统计出来。
        for (int c : C) {
            for (int d : D) {
                if (umap.find(0 - (c + d)) != umap.end()) {
                    count += umap[0 - (c + d)];
                }
            }
        }
        return count;
    }
};

时间复杂度O(n^2)。

383. 赎金信

题目链接
给定一个赎金信 (ransom) 字符串和一个杂志(magazine)字符串，判断第一个字符串 ransom 能不能由第二个字符串 magazines 里面的字符构成。如果可以构成，返回 true ；否则返回 false。
(题目说明：为了不暴露赎金信字迹，要从杂志上搜索各个需要的字母，组成单词来表达意思。杂志字符串中的每个字符只能在赎金信字符串中使用一次。)

注意：
你可以假设两个字符串均只含有小写字母。
canConstruct(“a”, “b”) -> false
canConstruct(“aa”, “ab”) -> false
canConstruct(“aa”, “aab”) -> true

重点

这道题和有效字母异位非常像，首先考虑用数组。

代码

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

15. 三数之和

题目链接
给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有满足条件且不重复的三元组。

注意：答案中不可以包含重复的三元组。
示例：
给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为： [ [-1, 0, 1], [-1, -1, 2] ]

重点

本题要求的三元组是不能重复的，用哈希法需要去重，比较麻烦。

双指针法：
要对数组进行排序，for循环确定a，left指针确定b，right指针确定c。

如果nums[i] + nums[left] + nums[right] > 0 就说明此时三数之和大了，因为数组是排序后了，所以right下标就应该向左移动，这样才能让三数之和小一些。
如果 nums[i] + nums[left] + nums[right] < 0 说明此时三数之和小了，left 就向右移动，才能让三数之和大一些，直到left与right相遇为止。这里就很能体现排序的好处。
有关去重
a如果重复了，应该直接跳过去，但这里有一个细节，就是判断 nums[i] 与nums[i + 1]是否相同，还是判断 nums[i] 与nums[i-1] 是否相同。如果是判断 nums[i] 与nums[i + 1]是否相同，那就把三元组中出现重复元素的情况直接pass掉了。例如{-1, -1 ,2} 这组数据，当遍历到第一个-1 的时候，判断下一个也是-1，那这组数据就pass了。
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) break;//a大于零，肯定不能三数和为零
            //if(i>0&&nums[i]==nums[i-1]) continue;//a的去重
            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]});
                    // 去重逻辑应该放在找到一个三元组之后，对b 和 c去重
                    while(left<right&&nums[left]==nums[left+1]) left++;
                    while(left<right&&nums[right]==nums[right-1]) right--;
                    left++;
                    right--;
                }
            }
            //把a的去重加到这里也行，这样abc的去重方法就统一了
            //但要注意i的取值范围，防止下标越界
            while(i<nums.size()-1&&nums[i]<=0&&nums[i]==nums[i+1]){
                i++;
            }           
        }
        return result;
    }
};

时间复杂度: O(n^2)

18. 四数之和

题目链接
题意：给定一个包含 n 个整数的数组 nums 和一个目标值 target，判断 nums 中是否存在四个元素 a，b，c 和 d ，使得 a + b + c + d 的值与 target 相等？找出所有满足条件且不重复的四元组。

注意：
答案中不可以包含重复的四元组。
示例：给定数组 nums = [1, 0, -1, 0, -2, 2]，和 target = 0。满足要求的四元组集合为： [ [-1, 0, 0, 1], [-2, -1, 1, 2], [-2, 0, 0, 2] ]

重点

三数之和的双指针解法是一层for循环num[i]为确定值，然后循环内有left和right下标作为双指针，找到nums[i] + nums[left] + nums[right] == 0。

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

那么一样的道理，五数之和、六数之和等等都采用这种解法。

代码

tips:
long sum=(long)(nums[a]+nums[b]+nums[left]+nums[right])会超出int范围，而long sum=(long)nums[a]+nums[b]+nums[left]+nums[right]不会，因为前者是把四个数先相加保存为int，再转为long，就超出了；而后者是把nums[a]变为long，long和int相加时会做整形提升，最终的和是long。

还有注意剪枝的处理：target是任意值。比如：数组是[-4, -3, -2, -1]，target是-10，不能因为-4 > -10而跳过。

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> result;
        sort(nums.begin(),nums.end());
        for(int a=0;a<nums.size();a++){
            //一级剪枝
            if (nums[a]>target && nums[a] >= 0) break;
            if(a>0&&nums[a]==nums[a-1]) continue;
            for(int b=a+1;b<nums.size();b++){
                //二级剪枝
                if((nums[a]+nums[b])>target&&(nums[a]+nums[b])>=0) break;              
                if(b>a+1&&nums[b]==nums[b-1]) continue;
                int left=b+1;
                int right=nums.size()-1;
                while(left<right){
                    // nums[k] + nums[i] + nums[left] + nums[right] > target 会溢出
                    //long sum=(long)(nums[a]+nums[b]+nums[left]+nums[right]);
                    long sum=(long)nums[a]+nums[b]+nums[left]+nums[right];
                    if(sum>target){
                        right--;
                    }
                    else if(sum<target){
                        left++;
                    }
                    else{
                        result.push_back(vector<int>{nums[a],nums[b],nums[left],nums[right]});
                        while(left<right&&nums[left]==nums[left+1]) left++;
                        while(left<right&&nums[right]==nums[right-1]) right--;
                        left++;
                        right--;
                    }
                }
            }
        }
        return result;
    }
};