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

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

454.四数相加II

题目链接：454.四数相加II
Given four integer arrays nums1, nums2, nums3, and nums4 all of length n, return the number of tuples (i, j, k, l) such that:

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

Example 1:

Input: nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2]
Output: 2
Explanation:
The two tuples are:
1. (0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0
2. (1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0

Example 2:

Input: nums1 = [0], nums2 = [0], nums3 = [0], nums4 = [0]
Output: 1

Constraints:

```clike
n == nums1.length
n == nums2.length
n == nums3.length
n == nums4.length
1 <= n <= 200
-228 <= nums1[i], nums2[i], nums3[i], nums4[i] <= 228

思路：
1.需要用unordered map存储a+b的值以及次数。
2.便利所有a+b和c+d的情况。

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {

        unordered_map<int,int> umap;
        int count = 0;
        for(int num1:nums1){
            for(int num2:nums2){
                umap[num1+num2]++;
            }
        }

        for(int num3:nums3){
            for(int num4:nums4){
                if (umap.find(-(num3+num4))!=umap.end()){
                    count+=umap[-(num3+num4)];
                };
            }
        }

        return count;
    }
};

注意 ：

1. 用a+b对应-c-d

383. 赎金信

题目链接：LeetCode 704 二分查找

思路：
1.用数组表示hashmap
2.先填充a，再用b减
3.b包含a，所以map的值必须<=0;

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        int vec[26]={0};

        if (ransomNote.size() > magazine.size()) {
            return false;
        }

        for (char r:ransomNote){
            vec[r-'a']++;
        }

        for (char m:magazine){
            vec[m-'a']--;
        }

        for (int v:vec){
            if (v>0){
                return false;
            }
        }
        return true;
    }
};

注意 ：

1. 注意字符串迭代为char a:A
2. 数组定义为int vec[26] = {0}

15. 三数之和

题目链接：15. 三数之和

思路：
1.双指针 i 和left、right。
2.遍历i，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 (left<right){
                if (nums[i]+nums[left]+nums[right] > 0){
                    right--; //大于0，移动right
                }
                else if (nums[i]+nums[left]+nums[right] < 0){
                    left++; //小于0，移动left
                }
                else{

                    result.push_back(vector<int>{nums[i],nums[left],nums[right]});
                    //等于0，保存结果
    
                    while(left<right && nums[left]==nums[left+1]){
                        left++;
                    }
                    while(left<right && nums[right]==nums[right-1]){
                        right--;
                    }
                    right --;
                    left ++;
                }
            }
        }
        return result;  
    }
};

注意 ：

1. i剪枝+去重，left和right同样减枝+去重；
2. vector {a,b,c}定义数组
3. 一定要在保存结果后去重

18. 四数之和

题目链接：18. 四数之和

思路：
1.思路与三数之和类似
2.多一个数的判断，剪枝+去重，每个数都需要

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        sort(nums.begin(),nums.end());
        vector<vector<int>> result;
        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 j = i+1; j<nums.size();j++){
                if (nums[i]+nums[j]>target && nums[i]+nums[j]>0){
                    break;
                    }
                if (j>i+1&&nums[j]==nums[j-1]){
                    continue;
                }
                int left = j+1;
                int right = nums.size()-1;
                
                while(left<right){
                    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++;
                    }

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

                }

            }
        }
        return result;
        
    }
};

注意 ：

1. i和j去重的时候判断 i>0 && nums[i] == nums[i-1]
2. left和right去重时判断nums[left]==nums[left+1] nums[right]==nums[right-1]