算法训练营Day07-LeetCode454. 四数相加 II& 383. 赎金信 & 15. 三数之和 && 18. 四数之和

454. 四数相加 II

题目链接：LeetCode454. 四数相加 II

给你四个整数数组 nums1 、nums2 、nums3 和 nums4 ，数组长度都是 n ，请你计算有多少个元组 (i, j, k, l) 能满足：

• 0 <= i, j, k, l < n
• nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0

思路：题目返回值为满足条件元组的个数，且数组元素值很大，应当想到使用哈希法，采用 map 结构。参考 LeetCode1. 两数之和 实现思路，将某个数组元素填入 map ，将剩余各数组元素加和与 map 中元素比对。总和为 0 ，返回值 cnt 加上 map 中 key 对应的 value 值。考虑到时间复杂度，map 存两个数组之和，如 nums1 nums2 。然后计算两个数组之和，如 nums3 nums4。时间复杂度为：O(n^2)+O(n^2) ，得到 O(n^2) 。

code：

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        int cnt = 0;
        unordered_map<int, int> map;

        for(int a : nums1){
            for(int b : nums2){
                map[a + b]++;
            }
        }

        for(int c : nums3){
            for(int d : nums4){
                int target = 0 - (c + d);
                if(map.find(target) != map.end()){
                    cnt += map[target];
                }
            }
        }
        return cnt;
    }
};

---

383. 赎金信

题目链接：LeetCode383. 赎金信

给你两个字符串：ransomNote 和 magazine ，判断 ransomNote 能不能由 magazine 里面的字符构成。

如果可以，返回 true ；否则返回 false 。

magazine 中的每个字符只能在 ransomNote 中使用一次。

思路：解法类似 LeetCode242.有效的字母异位词 。使用数组实现哈希法。先统计 magazine 字符串中各字母个数，在 record 中表现为字母对应位置元素值加一。再统计 ransomNote 字符串中各字母个数，在 record 中表现为字母对应位置元素值减一。最后判断 record 中是否有 < 0 的数组元素。

code：

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        int record[26] = {0};
        for(char i : magazine){
            record[i - 'a']++;
        }

        for(char j : ransomNote){
            record[j - 'a']--;
        }

        for(int num = 0; num < 26; num++){
            if(record[num] < 0){
                return false;
            }
        }

        return true;
    }
};

---

15. 三数之和

题目链接：LeetCode15. 三数之和

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

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

思考过程：通过先确定 a, b 的值，之后使用哈希法判断数组中是否存在 c 满足条件。但是结果集中包括重复三元组。使用双指针法同样能得到包含重复三元组的结果集，再次卡在去重操作。最终参考以下视频讲解，完成解题。

参考视频：梦破碎的地方！| LeetCode：15.三数之和

code：

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() - 2; i++){
            if(nums[i] > 0){
                break;
            }

            if(i > 0 && nums[i] == nums[i - 1]){
                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]});
                    while(right > left && nums[right] == nums[right - 1]){
                        right--;
                    }
                    while(right > left && nums[left] == nums[left + 1]){
                        left++;
                    }

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

总结：虽然想到双支针，但是去重逻辑理不清楚。

---

18. 四数之和

题目链接：LeetCode18. 四数之和

给你一个由 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

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

思路：实现方式与三数之和一样，使用双指针法在三数之和基础上嵌套一个循环。但是剪枝条件应做出变化，target 可能为负数，仅判断 nums[i] > target 不能完成剪枝操作，在遇到连续负数情况下可能漏解。

code：

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> result;
        if(nums.size() < 4){
            return result;
        }

        sort(nums.begin(), nums.end());
        for(int i = 0; i < nums.size() - 3; i++){
            if(nums[i] > target && nums[i] > 0){
                break;
            }
            if(i > 0 && nums[i] == nums[i - 1]){
                continue;
            }
            for(int j = i + 1; j < nums.size() - 2; j++){
                if(nums[i] + nums[j] > target && nums[j] > 0){
                    break;
                }
                if(j > i + 1 && nums[j] == nums[j - 1]){
                    continue;
                }

                int left = j + 1;
                int right = nums.size() - 1;
                while(right > left){
                    long sum = (long) nums[i] + nums[j] + nums[left] + nums[right]; 
                    if(sum > target){
                        right--;
                    }else if(sum < target){
                        left++;
                    }else{
                        result.push_back(vector<int>{nums[i], nums[j], nums[left], nums[right]});
                        while(right > left && nums[left] == nums[left + 1]){
                            left++;
                        }
                        while(right > left && nums[right] == nums[right - 1]){
                            right--;
                        }
                        left++;
                        right--;
                    }
                }
            }
        }
        return result;
    }
};

---

参考文章：

代码随想录-哈希表-454. 四数相加 II
代码随想录-哈希表-15. 三数之和
代码随想录-哈希表-18.四数之和