二刷代码随想录第七天

454. 四数相加 II

  • 先用map记录前两个数的和num1 + num2的值出现了多少次
  • 再在后两个数组里找0 - (num1 + num2),找到后就累加map中的次数
cpp 复制代码
class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3,
                     vector<int>& nums4) {
        unordered_map<int, int> map;
        for (int num1 : nums1) {
            for (int num2 : nums2) {
                map[num1 + num2]++;
            }
        }
        int count = 0;
        for (int num3 : nums3) {
            for (int num4 : nums4) {
                int need = 0 - (num3 + num4);
                if (map.find(need) != map.end()) {
	                //!!!!!!!!!这里是加上map[need]的次数,不是++
                    count+=map[need];
                }
            }
        }

        return count;
    }
};

383. 赎金信

cpp 复制代码
class Solution {
public:
    bool canConstruct(string ransomNote, string magazine) {
        vector<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 n : arr) {
            if (n > 0) {
                return false;
            }
        }
        return true;
    }
};

15. 三数之和

  • 首先对数组排序
  • 定义三个指针,一个i从0开始遍历数组,一个left在i右边一位,一个right在数组末尾
  • 确定i,不断移动leftright,同时要注意剪枝
  • nums[i] > 0,第一个都大于0了,那后面不管怎样也不可能等于0
  • i > 0 && nums[i] == nums[i - 1],重复元素就跳过,要找前一个元素,才是用过的元素
  • 找到三元数组收缩leftright时,也需要去重,用while去找,找到第一个不相等的元素
cpp 复制代码
class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ans;
        sort(nums.begin(), nums.end());
        for (int i = 0; i < nums.size(); i++) {
            if (nums[i] > 0) {
                return ans;
            }
            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) {
                    right--;
                } else if (nums[i] + nums[left] + nums[right] < 0) {
                    left++;
                } else {
                    ans.push_back({nums[left], nums[right], nums[i]});
                    while (left < right && nums[left] == nums[left + 1]) {
                        left++;
                    }
                    while (left < right && nums[right] == nums[right - 1]) {
                        right--;
                    }
                    left++;
                    right--;
                }
            }
        }

        return ans;
    }
};

18. 四数之和

  • 同上
  • 注意是和target做比较了
  • 数据太大需要加一个(long)
cpp 复制代码
class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> ans;
        sort(nums.begin(), nums.end());
        for (int i = 0; i < nums.size(); i++) {
            if (nums[i] >= 0 && nums[i] > target) {
                return ans;
            }
            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 ((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 {
                        ans.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 ans;
    }
};
相关推荐
大数据张老师3 小时前
数据结构——邻接矩阵
数据结构·算法
低音钢琴3 小时前
【人工智能系列:机器学习学习和进阶01】机器学习初学者指南:理解核心算法与应用
人工智能·算法·机器学习
傻童:CPU5 小时前
C语言需要掌握的基础知识点之前缀和
java·c语言·算法
又见野草6 小时前
软件设计师知识点总结:数据结构与算法(超级详细)
数据结构·算法·排序算法
GalaxyPokemon6 小时前
有一个服务器,用于提供HTTP服务,但是需要限制每个用户在任意的100秒内只能请求60次,怎么实现这个功能
算法
fl1768316 小时前
基于opencv+Mediapipe+CNN实现用手势识别控制对鼠标操控python源码+项目说明+设计文档
算法
K 旺仔小馒头6 小时前
优选算法:01 双指针巧解移动零问题
c++·算法·刷题
sali-tec7 小时前
C# 基于halcon的视觉工作流-章49-网面破损
开发语言·图像处理·算法·计算机视觉·c#
ysa0510307 小时前
Fenwick 树进行快速统计
算法
im_AMBER8 小时前
Leetcode 33
算法·leetcode·职场和发展