【代码随想录算法训练营第七天| 454.四数相加II、383. 赎金信、15. 三数之和、18. 四数之和】

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

454.四数相加II

老样子,先看代码:

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int,int> cnt;
        for(auto n3:nums3)
            for(auto n4:nums4)
                cnt[n3+n4]++;
        int res = 0;
        for(auto n1:nums1)
            for(auto n2:nums2)
                res += cnt[-(n1+n2)];
        return res;
    }
};

这道题目和前面letcoded第一题两数之和差不多,但是也是相对麻烦一点,就是这里是四个数组,首先说写一个四重循环行不行,n的范围是1~200,四重循环就是1.6 * 109 这样会超时,所以最好降到n2 的时间复杂度,所以我们可以两个数组,两个数组的看,首先看nums3和nums4,先计算这两个数组中所有数字组合相加的和,然后用一个哈希表存储每个和出现的次数,对于一个n1+n2,有多少个-(n3+n4)在,就有多少种可能,这样就计算出满足条件的元组个数了。

383. 赎金信

看代码:

class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        unordered_map<char,int> cnt;
        for(auto m:magazine)
            cnt[m]++;
        for(auto r:ransomNote)
        {
            cnt[r]--;
            if(cnt[r] < 0)
                return false;
        }
        return true; 
    }
};

这道题目说看看magazine中的字符能否组成ransomNote,所以只要保证magazine中有ransomNote的出现的所有字母,而且出现的次数大于等于ransomNote中就可以。所以我们先对magazine中字母进行统计,然后对ransomNote进行检查,相当于ransomNote在使用magazine的字符,magazine中的字符用一个少一个,所以就进行"–"操作,如果次数小于0说明magazine中该字符数量不够或者根本就没有。那就返回false。

15. 三数之和

先看代码:

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
       vector<vector<int>> res;
       sort(nums.begin(),nums.end());
       for(int i = 0;i < nums.size();i++)
       {
           if(i && nums[i] == nums[i - 1]) continue;
           for(int j = i + 1,k = nums.size() - 1;j < k;j++)
           {
               if(j > i+1 && nums[j] == nums[j - 1]) continue; //j不能和i重叠
               while(j < k - 1 && nums[i] + nums[j] + nums[k - 1] >= 0) k--;  //如果k的下一个数和j没有重叠
               if(nums[i] + nums[j] + nums[k] == 0)
               {
                   res.push_back({nums[i],nums[j],nums[k]});
               }
           }
       }
       return res;
    }
};

这道题目用双指针好一些,我用哈希表试着做了,有点麻烦,所以这里就说双指针吧,数组用双指针,需要数组是有序的,所以首先上来排序,然后是思路,首先我先定下i,然后对j和k进行移动,i < j < k,三者不能有重叠,j从左往右走,k从右往左走,只要相遇就停下,这里我们可以观察这个式子:nums[i] + nums[j] + nums[k],现在数组是有序的,nums[i]是固定的,所以当j向右走的时候,nums[j]就在增大,同时为了保证nums[i] + nums[j] + nums[k] = 0,nums[j]在增大,也就意味着nums[k]就要减小,所k从右往左走,当我nums[i] 和nums[j]固定了的时候(也就是while()循环里的式子),nums[k]是在减小的,每次我们尝试减小nums[k],只要nums[i] + nums[j] + nums[k - 1],也就是k的下一个数仍然满足条件,就继续,直到ums[i] + nums[j] + nums[k - 1] < 0,这个时候nums[i] + nums[j] + nums[k]不是0,那也是距离0最近的一个正数了,我们只要等于0的即可。
说不好,建议看看y总视频:

https://www.acwing.com/activity

18. 四数之和

看代码:

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        sort(nums.begin(),nums.end());
        vector<vector<int>> res;
        for(int i = 0;i < nums.size();i++)
        {
            if(i && 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;
                for(int k = j + 1,u = nums.size() - 1;k < u;k++)
                {
                    if(k > j+1 && nums[k] == nums[k - 1]) continue;
                    while(k < u - 1 && (long)nums[i] + nums[j] + nums[k] + nums[u - 1] >= target) u--;
                    if(((long)nums[i] + nums[j] + nums[k] + nums[u]) == target)
                        res.push_back({nums[i],nums[j],nums[k],nums[u]});
                }
            }
        }
        return res;
    }
};

原理和三个数之和一样,这次是i,j,k在左边,u从右往左走,不过这个题的测试用例是真坑啊,真坑啊,真坑啊,注意中间的long类型。

  • 18
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值