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

yyqx_pyt

已于 2024-06-15 17:13:55 修改

阅读量261

点赞数 4

文章标签：算法数据结构哈希表

于 2024-06-15 16:24:16 首次发布

本文链接：https://blog.csdn.net/yyqx_where/article/details/139703087

版权

第454题.四数相加II

力扣题目链接(opens new window)

给定四个包含整数的数组列表 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]

输出:

解释:

两个元组如下:

(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

本题需要找到满足于A[i] + B[j] + C[k] + D[m]=0的组合数，即A[i] + B[j] = -（C[k] + D[m]），所以我们可以求出数组A和数组B的元素之和，存在哈希表中，然后在哈希表中查找是否存在-（C[k] + D[m]）。

首先，定义count表示返回的数量，新建哈希表unordered_map<int,int>map，其中key是指a+b的值，value是指a+b出现的次数，遍历数组A和数组B，a+b每出现一次就++；

之后，遍历数组C和数组D,在map中查找-（c+d）是否出现，这里使用find函数，出现的话，将count与出现的次数相加。这里count不能直接++，因此-（c+d）可能会出现多次的情况，产生的组合也会不同

最后返回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){
            for (int b: nums2){
                map[a+b]++;
            }
        }
        int count = 0;
        for (int c:nums3){
            for (int d: nums4){
                if (map.find(-(c+d)) != map.end()){
                    count += map[-(c+d)];//不能直接++，要注意出现的次数
                }
            }
        }
        return count;
    }
};

383. 赎金信

给你两个字符串：ransomNote 和 magazine ，判断 ransomNote 能不能由 magazine 里面的字符构成。如果可以，返回 true ；否则返回 false 。magazine 中的每个字符只能在 ransomNote 中使用一次。

示例 1：

输入：ransomNote = "a", magazine = "b"
输出：false

示例 2：

输入：ransomNote = "aa", magazine = "ab"
输出：false

示例 3：

输入：ransomNote = "aa", magazine = "aab"
输出：true

换汤不换药的一道题，如果ransomNote可以由magezine里面的字符构成，说明ransomNote里出现的字符在magazine里都出现过，所以我们只需要在数组map记录magezine里字符出现的次数，然后遍历ransomNote的字符，将出现在ransomNote的字符次数减一。最后观察数组map里是否存在次数小于0的字符，若有，说明ransomNote里存在的字符并没有出现在magezine中，即不能由magazine构成。

具体代码

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

15. 三数之和

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请你返回所有和为 0 且不重复的三元组。

注意：答案中不可以包含重复的三元组。

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
解释：
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意，输出的顺序和三元组的顺序并不重要。

示例 2：

输入：nums = [0,1,1]
输出：[]
解释：唯一可能的三元组和不为 0 。

示例 3：

输入：nums = [0,0,0]
输出：[[0,0,0]]
解释：唯一可能的三元组和为 0 。

这里采用的是双指针的方法，先对数组nums进行从小到大的排序，然后进行遍历，从i=0开始，然后定义指针left = i+1，right = nums.size(）-1,；

当left < right时候，如果nums[i] + nums[left] + nums[right] > 0,说明值偏大，right要左移，right--

如果nums[i] + nums[left] + nums[right] < 0,说明值偏小，left要右移，left++

如果nums[i] + nums[left] + nums[right] =0，将nums[i] ， nums[left] ， nums[right] 存入数组中

题目中指明需要进行去重操作

首先，nums[i]如果大于0，说明后面的值都是大于0，因此不存在nums[i] + nums[left] + nums[right]=0的情况，直接返回空数组

其次，当i>0时，如果nums[i]==nums[i-1]，说明存在相同的情况，直接跳出本次循环continue

最后，当找到nums[i] + nums[left] + nums[right] =0时候，判断nums[left]是否等于nums[left+1]，nums[right]是否等于nums[right-1]，如果相等，将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;
            }
            int left = i + 1;
            int right = nums.size() - 1;
            while (left < right){
                if (nums[i] + nums[left] + nums[right] < 0){
                    left++;
                }
                else if(nums[i] + nums[left] + nums[right] > 0){
                    right--;
                }else{
                    result.push_back(vector<int>{nums[i], nums[left], nums[right]});
                    while (right > left && nums[right] == nums[right - 1]) right--;//left和right去重
                    while (right > left && nums[left] == nums[left + 1]) left++;
                    right--;
                    left++;
                }
            }
        }
        return result;
    }
};

18. 四数之和

给你一个由 n 个整数组成的数组 nums ，和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] （若两个四元组元素一一对应，则认为两个四元组重复）：

0 <= a, b, c, d < n
a、b、c 和 d 互不相同
nums[a] + nums[b] + nums[c] + nums[d] == target

你可以按 任意顺序 返回答案。

示例 1：

输入：nums = [1,0,-1,0,-2,2], target = 0
输出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]

示例 2：

输入：nums = [2,2,2,2,2], target = 8
输出：[[2,2,2,2]]

这道题在三数之和的基础上，再加上了一层循环，遍历两次数组nums，当然需要先做排序，第一层for循环，从i= 0开始，第二层for循环从j=i+1开始，然后定义指针left = j +1，right = nums.size()-1，之后的判断方法和上一题一样，只是多加了一个元素。

关于去重，要多考虑一个元素，并且要进行元素之和和target的比较，并且题目中已说数组元素都是大于0的，

具体代码：

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

            for (int k = i + 1; k < nums.size(); k++){
                if (nums[i] + nums[k] > target && nums[i] + nums[k] >= 0){
                    break;
                }
                if (k > i + 1 && nums[k] == nums[k - 1]){
                    continue;
                }
                int left = k + 1;
                int right = nums.size() - 1;
                while (left < right){
                    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 (left < right && nums[left] == nums[left + 1]) left++;
                        while (left < right && nums[right] == nums[right - 1]) right--;

                        right--;
                        left++;
                    }
                }
            }
        }
        return result;
    }
};

yyqx_pyt

关注

4
点赞
踩
5

收藏

觉得还不错? 一键收藏
0
评论
代码随想录算法训练营第七天| ● 454.四数相加II ● 383. 赎金信 ● 15. 三数之和 ● 18. 四数之和

本题需要找到满足于A[i] + B[j] + C[k] + D[m]=0的组合数，即A[i] + B[j] = -（C[k] + D[m]），所以我们可以求出数组A和数组B的元素之和，存在哈希表中，然后在哈希表中查找是否存在-（C[k] + D[m]）。给定四个包含整数的数组列表 A , B , C , D ,计算有多少个元组 (i, j, k, l) ，使得 A[i] + B[j] + C[k] + D[l] = 0。
复制链接

扫一扫