训练营day7|454.四数相加II、383. 赎金信、15. 三数之和、18. 四数之和

454.四数相加II

• 题目链接：454.四数相加II
• 独立思路：❌ 没想出来（四层循环。。。）
• 题解：首先求出 A 和 B 任意两数之和 sumAB，以 sumAB 为 key，sumAB 出现的次数为 value，存入 hashmap 中。然后计算 C 和 D 中任意两数之和的相反数 sumCD，在 hashmap 中查找是否存在 key 为 sumCD。
• 时间复杂度：O(n2)
• 空间复杂度：O(n2)

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        int n = nums1.size();   
        int ans = 0;

        map<int, int> mp;   //key值存nums1\2之和，value值存该值出现过的次数
        for(int i = 0; i < n; i++){
            for(int j = 0; j < n; j++){
                int sum12 = nums1[i] + nums2[j];
                mp[sum12]++;
            }
        }

        for(int k = 0; k < n; k++){
            for(int l = 0; l < n; l++){
                int sum34 = -(nums3[k] + nums4[l]);
                if(mp.find(sum34) != mp.end()){
                    auto s = mp.find(sum34);
                    ans += s->second;
                }
            }
        }

        return ans;
    }
};

383. 赎金信

• 题目链接：383. 赎金信
• 独立思路：用map存储在magazine中出现过的字符及其次数，再遍历ransomNote字符，每用一个字符value--
• 时间复杂度：O(m+n)
• 空间复杂度：O(n)

//想法：用map存储在magazine中出现过的字符及其次数，再遍历ransomNote字符，每用一个字符value--
class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        unordered_map<char, int> mag;
        for(char ch1: magazine){
            mag[ch1]++;
        }

        for(char ch2 : ransomNote){
            auto it = mag.find(ch2);
            if(it != mag.end() && it->second > 0){   // 字符在magzine中存在而且次数还没用完
                it->second--;
            }else
                return false;
        }

        return true;
    }
};

15. 三数之和（没很懂，再多看）

• 题目链接：15. 三数之和
• 独立思路：❌ 没想出来，理解成了返回{i, j, k }
• 题解：（排序+双指针）先对数组进行排序，然后遍历数组，对于每个元素，使用两个指针在其后面的数组中寻找两个数，使得它们的和加上当前元素的和为零。同时，通过跳过重复的元素来避免重复的三元组。
• 时间复杂度：O(n2)
• 空间复杂度：O(1)

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ans;
        sort(nums.begin(), nums.end());
        int n = nums.size();

        for (int i = 0; i < n; i++) {
            if (i > 0 && nums[i] == nums[i - 1]) 
                continue; // 跳过重复元素

            int j = i + 1, k = n - 1;
            while (j < k) {
                int sum = nums[i] + nums[j] + nums[k];
                if (sum == 0) {
                    ans.push_back({nums[i], nums[j], nums[k]});
                    while (j < k && nums[j] == nums[j + 1]) j++; // 跳过重复元素
                    while (j < k && nums[k] == nums[k - 1]) k--; // 跳过重复元素
                    j++; k--;
                } else if (sum < 0) {
                    j++;
                } else {
                    k--;
                }
            }
        }
        return ans;
    }
};

18. 四数之和

• 题目链接：18. 四数之和
• 独立思路： 与三数之和相似，先将nums数组排序，a从0开始，b从a+1开始，d从n-1开始，c从d-1开始。两个for循环来记录a和d的变化。b和c的变化在while循环中实现。
• 问题(已解决)：用int类型声明sum会溢出！要用long而且还得进行强制类型转换
• 时间复杂度：O(n3)
• 空间复杂度：O(log2n) //排序算法的

//想法：与三数之和相似，先将nums数组排序，a从0开始，b从a+1开始，d从n-1开始，c从d-1开始
class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> ans;
        int n = nums.size();
        sort(nums.begin(), nums.end());

        for(int a = 0; a < n-3; a++){
            if(a > 0 && nums[a] == nums[a-1])   //外指针a排除重复元素
                    continue;
            for(int d = n-1; d > 2; d--){
                if(d < n-1 && nums[d] == nums[d+1]) //外指针d排除重复元素
                    continue;

                int b = a+1; int c = d-1;
                while(b < c){
                    long sum = (long)nums[a] + nums[b] + nums[c] + nums[d];    //int会溢出
                    if(sum == target){
                        ans.push_back({nums[a], nums[b], nums[c], nums[d]});
                        while(b < c && nums[b] == nums[b+1])    //内指针b排除重复元素
                            b++;
                        while(b < c && nums[c] == nums[c-1])    //内指针c排除重复元素
                            c--;
                        b++; c--;
                    }else if(sum < target){
                        b++;
                    }else{
                        c--;
                    }
                }//while
            }//小for
        }//大for

        return ans;
    }
};