代码随想录算法训练营第七天| LeetCode454 四数相加II、LeetCode383 赎金信、LeetCode15 三数之和、LeetCode18 四数之和

454.四数相加II 

思路:遍历前两个数组,将前两个数组的和制成哈希表,key值为前两个数组元素的和,value值为出现的次数。遍历后两两个数组,求后两个数组对应的0-(c+d),在构造的哈希表中寻找,如果能找的到,计数值加上对应的value值。遍历结束后返回计数值。

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        //遍历1、2数组,构建哈希表
        unordered_map<int,int> map;
        int count = 0;
        for(int i = 0;i<nums1.size();i++)
        {
            for(int j = 0;j<nums2.size();j++)
            {
                map[nums1[i]+nums2[j]]++;
            }
        }
        for(int a =0;a<nums3.size();a++)
        {
            for(int b=0;b<nums4.size();b++)
            {
                 int target = 0-( nums3[a]+nums4[b]);
                if(map.find(target)!=map.end())
                {
                    count+= map[target];
                }
            }
        }
        return count;
    }
};

383. 赎金信 

思路:将magazine中的字符利用数组存为哈希表,记录字符的个数,根据之前一道题的思路,小写字母有26位,构建一个int hash[26];然后遍历ransomNote中的字符,如果在哈希表中,则字符对应的个数-1。遍历结束后,如果哈希表中,存在小于0的数,则说明false,反之true。

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

15. 三数之和

思路:首先要排序,利用双指针的思路,分别定义i,left,right,来寻找满足条件的三元组,重点主要是去重和剪枝。对i进行去重时,i>0条件下,如果nums[i]=nums[i-1];continue;对i进行剪枝操作,如果nums[i]>target,由于我们排序后数组时从大到小排列的,因此i后面的元素更不能一起求和等于0了。对left和right进行去重操作,满足left<right的条件下,如果nums[right]=nums[right-1];则right--;left同理。

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

18. 四数之和 

思路:相比于三数之和,外层多了一个循环,内部循环同理,重要的是对k,i,left,right进行去重和剪枝应该如何操作。第一布仍需要对数组进行排序,但这个时候target的值可能为正也可能为负,因此在剪枝操作时,判断改为了nums[k]>target&&nums[k]>0时,break; 内层的同理 判断nums[k]+nums[i]与target的关系。注意内部i去重时,要在i>k+1的条件下,要不找不到nums[i-1]。

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

        }
            return result;
    }
};

收获:

1、学会了用双指针法处理数组中多个元素求和的问题。

2、求和问题,先对数组进行排序,然后要注意题目条件,进行去重和剪枝。

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值