代码随想录训练营day7|454.四数相加II,15. 三数之和,18. 四数之和

插入map的两种方法

map.insert(pair<int,int>(key,value));//first
map[key]=value;//second

454.四数相加2

四数相加II
思路:
将数组两两分组,把两组数组的元素相加有多少情况算出来,可以用map来记录,然后查找target-sum是否在map中出现过

unordered_map <int,int> mp;
int ans=0;
for(int n1:nums1){
    for(int n2:nums2){
        int sum=n1+n2;
        if(mp.find(sum)!=mp.end()){
            mp[sum]++;//map的value可以修改
        }
        else{
            mp.insert(pair<int,int>(sum,1));
        }
    
    }
}

然后在剩下两组计算时查找

for(int n3:nums3){
	for(int n4:nums4){
	      int s=-(n3+n4);
	      if(mp.find(s)!=mp.end()){
	          ans+=mp[s];
	      }
	    }
	}
	return ans;

15.三数之和

思路:固定一个数,剩下两个数依次查找他们的和看与固定过的是否相加等于target。
为了避免重复查找可以排序。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        //先给nums排序
        vector<vector<int>> result;
        for(int i=0;i<nums.size();i++){
            if(i>0&&nums[i]==nums[i-1]){
                continue;
            }//往前查重而非往后查重,因为left就在i后面,可能1,1,-2的答案会被排掉
            int left=i+1,right=nums.size()-1;
            while(left<right){//不用=因为要找三个数
                int sum=nums[i]+nums[left]+nums[right];
                if(sum>0)
                    right--;//因为已经排过序,右边最大左边最小
                else if(sum<0)
                    left++;
                else{
                    result.push_back({nums[i],nums[left],nums[right]});
                    while(left<right&&nums[right]==nums[right-1]){
                        right--;//如果前一个和right一样则前移
                    }
                    while(left<right&&nums[left]==nums[left+1]){
                        left++;
                    }
                    right--;
                    left++;
                }
            }
        }
        return result;
    }
};

18.四树之和

18.四树之和
和三数之和是一样的,只是再嵌套了一层循环

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,right=nums.size()-1;
                while(left<right){
                    long long int sum=(long)nums[i]+nums[j]+nums[left]+nums[right];
                    if(sum>target){
                        right--;
                    }
                    else if(sum<target){
                        left++;
                    }
                    else{
                        result.push_back({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;
    }
};

疑惑

for (int first = 0; first <= nums.size()-4;first++) 
....
for(int second = first + 1;second<=nums.size()-3;second++)

当我其他不改动,只是边界写成这样时,
测试数组[0]会出现越界的问题,
探索半天后只能在最前面判断nums.size()<4,小于的话直接return

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值