*算法训练(leetcode)第六天 | 454. 四数相加 II、383. 赎金信、15. 三数之和、18. 四数之和

454. 四数相加 II

leetcode题目地址

拿到题目最初的思路就是四重循环暴力求解。从代码随想录的介绍了解了使用哈希表的解法。

先遍历两个数组,将其中任意两值相加存入哈希表中,再遍历剩余两个数组,将其中任意两值相加,在哈希表中查找当前两值之和的相反数的个数。将这些个数加起来即为所求结果。

时间复杂度: O ( n 2 ) O(n^{2}) O(n2)
空间复杂度: O ( n 2 ) O(n^{2}) O(n2)

// c++
class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int, int> hash;
        int result = 0;
        for(int a : nums1){
            for(int b : nums2){
                hash[a+b]++;
            }
        }

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

383. 赎金信

leetcode题目地址

创建哈希表记录magazine中每个字符的个数,遍历ransomNote中的字符,每个字符对应哈希表的值减一,若哈希表的值出现负值则返回false。

写法一:

使用unordered_map做哈希表。

时间复杂度: O ( n ) O(n) O(n)
空间复杂度: O ( n ) O(n) O(n)

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

写法二:

使用一维数组做哈希表。

时间复杂度: O ( n ) O(n) O(n)
空间复杂度: O ( 1 ) O(1) O(1)

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

使用map的空间开销大于数组,因为map要维护红黑树或者哈希表。而且还要做哈希函数,也是费时的。

15. 三数之和

leetcode题目地址

循环遍历的同时外加双指针法两侧向中间逼近寻找三值和为0的三元组。

时间复杂度: O ( n 2 ) O(n^{2}) O(n2)
空间复杂度: O ( 1 ) O(1) O(1)

// c++
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){
                    result.emplace_back(vector<int>{nums[i], nums[left], nums[right]});
                    while(left < right && nums[left]==nums[left+1]) left++;
                    while(left < right && nums[right] == nums[right-1]) right--;
                    left++;
                    right--;
                }else if(nums[i] + nums[left] + nums[right] > 0) right--;
                else left++;
            }
        }
        return result;
    }
};

18. 四数之和

leetcode题目地址

和上题三数之和有异曲同工之妙,外层多加一层循环既可。难点在于去重。

时间复杂度: O ( n 3 ) O(n^{3}) O(n3)
空间复杂度: O ( 1 ) O(1) O(1)

// c++
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 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(nums[i] + nums[j] + nums[left] + nums[right] > target) right--;
                    else if(nums[i] + nums[j] + nums[left] + nums[right] < target) left++;
                    else{
                        result.emplace_back(vector<int>{nums[i], nums[j], nums[left], nums[right]});
                        while(left<right && nums[left]==nums[left+1]) left++;
                        while(left<right && nums[right]==nums[right-1]) right--;
                        left++;
                        right--;
                    }
                    
                }
               
               
            }
        }
        return result;
    }
};
  • 21
    点赞
  • 17
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值