代码随想录第七天|454.四数相加、383.赎金信、15.三数之和、18、四数之和

最新推荐文章于 2024-10-04 20:45:38 发布

xiaoning__

最新推荐文章于 2024-10-04 20:45:38 发布

阅读量182

点赞数

文章标签：算法

本文链接：https://blog.csdn.net/xiaoning__/article/details/131184022

版权

本文介绍了几个基于数组和哈希表的算法问题，包括四数相加求解元组数量、赎金信判断字符构成、三数之和的寻找以及四数之和的解法，特别强调了哈希表在优化搜索效率和处理重复元素中的作用，以及双指针技术在处理数组问题中的应用。此外，还讨论了在四数之和问题中引入剪枝策略来提高性能。

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

题目

454.四数相加

给定四个包含整数的数组列表 A , B , C , D ,计算有多少个元组 (i, j, k, l) ，使得 A[i] + B[j] + C[k] + D[l] = 0。

首先需定义一unordered_map，遍历前两个数组，求两数组元素之和，将两数之和放入key中，两数之和出现的次数放入value中。定义resul记录满足的个数。

接着遍历另两数组，若此两数组之和的相反数出现在前两数组中，将value加到result中。

最后返回result。

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        //错误版本
        /*int len1=nums1.size()*nums2.size();
        int len2=nums3.size()*nums4.size();
        vector<int> nums5(len1,0);
        vector<int> nums6(len2,0);
        int k=0;
        for(int i=0;i<nums1.size();i++) {
            for(int j=0;j<nums2.size();j++) {
                nums5[k++]=nums1[i]+nums2[j];
            }
        }
        for(int i=0;i<nums3.size();i++) {
            for(int j=0;j<nums4.size();j++) {
                nums5[k++]=nums3[i]+nums4[j];
            }
        }
        multiset<int> record(nums5.begin(),nums5.end());
        int result=0;
        for(int i=0;i<nums6.size();i++) {
            if(record.find(-nums6[i])!=record.end()) {  //错误原因是如果执行这步resul+1，但
                result++;                               //实际上A+B等于该数的不只有一个
            }
        }
        return result;*/
        unordered_map<int,int> map;
        for(int num1:nums1) {
            for(int num2:nums2) {
                map[num1+num2]++;
            }
        }
        int result=0;
        for(int num3:nums3) {
            for(int num4:nums4) {
                if(map.find(-(num3+num4))!=map.end()) {
                    result+=map[-(num3+num4)];
                }
            }
        }
        return result;
    }
};

383.赎金信

给定一个赎金信 (ransom) 字符串和一个杂志(magazine)字符串，判断第一个字符串 ransom 能不能由第二个字符串 magazines 里面的字符构成。如果可以构成，返回 true ；否则返回 false。

可用map或者数组解决该问题。

map：

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        //map
        unordered_map<char,int> map;
        for(int i=0;i<magazine.size();i++) {
            map[magazine[i]]++;
        }
        for(int j=0;j<ransomNote.size();j++) {
            if(map.find(ransomNote[j])!=map.end()) {
                map[ransomNote[j]]--;
                if(map[ransomNote[j]]<0) return false;
            }
            else return false;
        }
        return true;
    }
};

数组：

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        //数组
        int result[26]={0};
        if(ransomNote.size()>magazine.size()) return false;  //若赎金信的字符数大于杂志，肯定构不成
        for(int i=0;i<magazine.size();i++) {
            result[magazine[i]-'a']++;
        }
        for(int i=0;i<ransomNote.size();i++) {
            result[ransomNote[i]-'a']--;
        }
        for(int i=0;i<26;i++) {
            if(result[i]<0) return false;
        }
        return true;
    }
};

15.三数之和

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

可利用双指针进行操作。

首先将数组进行排列，有一层for循环，i从下标0的地方开始（代表第一个元素），同时需定义一个left=i+1（代表第二个元素）和一个right在数组的结尾（代表第三个元素）。

接着进行第一个元素去重操作，若该元素和上一次循环的数相等，则跳出循环。

如果nums[i]+nums[left]+nums[right]>0，则移动right向前，如果nums[i]+nums[left]+nums[right]<0，则移动left向后。如果nums[i]+nums[left]+nums[right]=0，则将答案放入result。同理也需对第二第三个元素进行去重操作。当left与它下一个元素相等，left向后移动一个，当right与它上一个元素相等，right向前移动一个。之后left与right分别向中间移动一步。

最后返回result。

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(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++;
                    left++;
                    right--;
                }
                
            }
        }
        return result;
    }
};

18.四数之和

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

与三数之和相似吗，区别在于多了一层for循环。

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

加入剪枝：

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> result;
        sort(nums.begin(), nums.end());
        for (int k = 0; k < nums.size(); k++) {
            // 剪枝处理
            if (nums[k] > target && nums[k] >= 0) {
            	break; // 这里使用break，统一通过最后的return返回
            }
            // 对nums[k]去重
            if (k > 0 && nums[k] == nums[k - 1]) {
                continue;
            }
            for (int i = k + 1; i < nums.size(); i++) {
                // 2级剪枝处理
                if (nums[k] + nums[i] > target && nums[k] + nums[i] >= 0) {
                    break;
                }

                // 对nums[i]去重
                if (i > k + 1 && nums[i] == nums[i - 1]) {
                    continue;
                }
                int left = i + 1;
                int right = nums.size() - 1;
                while (right > left) {
                    // nums[k] + nums[i] + nums[left] + nums[right] > target 会溢出
                    if ((long) nums[k] + nums[i] + nums[left] + nums[right] > target) {
                        right--;
                    // nums[k] + nums[i] + nums[left] + nums[right] < target 会溢出
                    } else if ((long) nums[k] + nums[i] + nums[left] + nums[right]  < target) {
                        left++;
                    } else {
                        result.push_back(vector<int>{nums[k], nums[i], nums[left], nums[right]});
                        // 对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;
    }
};