代码随想录第七天:

454. 四数相加 II

复制代码
class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int,int> mp;
        for(int i=0;i<nums1.size();i++)
        {
            for(int j=0;j<nums2.size();j++)
            {
                mp[nums1[i]+nums2[j]]++;
            }
        }
        int count=0;
        for(int i=0;i<nums3.size();i++)
        {
            for(int j=0;j<nums4.size();j++)
            {
                auto it=mp.find(0-(nums3[i]+nums4[j]));
                if(it!=mp.end())
                {
                    count=count+mp[0-(nums3[i]+nums4[j])];
                }
            }
        }
        return count;
    }
};

本题如果采用暴力解法遍历四个数组的时间复杂度为n的四次方,但是通过map先计算前两个数组的值的出现次数,再遍历后两个数组,通过寻找map中中是否出现相应的差值,完成统计计算。

​​​​​​383. 赎金信

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

15. 三数之和

复制代码
class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        vector<vector<int>> arr;
        for(int i=0;i<nums.size();i++)
        {
            if(nums[i]>0) return arr;
            //a的去重
            if(i>0&&nums[i-1]==nums[i]) 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{
                    arr.push_back({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--;
                }
                
            }
        }
        return arr;
    }
};

采用双指针的思路,主要的关键是去重

18. 四数之和

复制代码
class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> arr;
        sort(nums.begin(),nums.end());
        for(int i=0;i<nums.size();i++)
        {
            if(nums[i]>target&&target>=0)
            {
                break;
            }
            if(i>0&&nums[i-1]==nums[i])
            {
                continue;
            }
            for(int j=i+1;j<nums.size();j++)
            {
                if(nums[i]+nums[j]>target&&target>=0)
                {
                    break;
                }
                if(j>i+1&&nums[j-1]==nums[j])
                {
                    continue;
                }
                int left=j+1;
                int right=nums.size()-1;
                while(left<right)
                {
                    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{
                        arr.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 arr;
    }
};

注意要用不满足条件的时候要用break,不要直接return,否则会错过多种情况。

相关推荐
倒头就睡的小比特19 小时前
算法竞赛C++常用的STL
c++·算法
weilx123419 小时前
C++笔记-文件IO-<fcntl.h>
c++
小羊没烦恼!20 小时前
初探性能优化——2个月到4小时的性能提升
java·开发语言·windows·算法·c#
猎头南楼21 小时前
知识社区推荐系统实践:新用户冷启动与长短期兴趣建模的挑战 资深推荐算法工程师
人工智能·深度学习·算法·机器学习
Smileyqp沛沛21 小时前
前端?C++ ?较大差异基础罗列
c++·基础·前端转c++
m0_5474866621 小时前
《数据结构教程》全套 PPT课件2026
数据结构
C语言小火车21 小时前
C/C++ 为什么需要编译器?
开发语言·c++
旖旎夜光1 天前
力控面试题 01.01: 判定字符是否唯一(位运算) —— 题解
c++·学习·算法·leetcode·力控
wzdark1 天前
大规模并行计算中的负载均衡算法研究4
算法
吞下星星的少年·-·1 天前
C++ 萌新语法入门篇
c++·算法比赛