15. 3Sum 找到数组中三数之和为给定值的组合

找到数组中三数之和为给定值的组合。

Given array nums = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

思考:求解三数之和,会先联想到求解两数之和。先来回忆一下两数之和怎么求解。方法1,使用unordered_map存放数组值和下标,再遍历数组从寻找是否又差值,有就找到一个组合。使用unordered_map存放下标,是为了这种case考虑,如sum=2, 数组中有一个1, 这时,再找就仍然找的是这个1就不对了。方法2. 在遍历是向set中存放没有配对的数值,如果可以从set中找到,那就找到一个组合了。这种方法不用存放下标。方法3.对原有数组进行排序,在用两个指针,low和high指针。如果它们的和大于sum,就high--,反之,low++。 用这三种方法进行拓展都可以解决3-sum问题。不过使用第三种方法最好。

解法:

    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 > 0 && nums[i] == nums[i - 1]) { continue; }
            
            int low = i + 1;
            int high = nums.size() - 1;
            
            while (low < high) {
                int three_sum = nums[i] + nums[low] + nums[high];
                if (three_sum == 0) {
                    res.push_back({nums[i], nums[low], nums[high]});
                    ++low;
                    --high;
                    
                    // 注意防止重复的结果
                    while (low < high && nums[low] == nums[low - 1]) { ++low; }
                    while (low < high && nums[high] == nums[high + 1]) { --high; }
                } else if (three_sum < 0) {
                    ++low;
                } else {
                    --high;
                }
            }
        }
        
        return res;
    }

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
给定一个整数数组 nums 和一个目标 target,要求在数组中找出个数的和等于目标,并返回这两个数的索引。 思路1:暴力法 最简单的思路是使用两层循环遍历数组的所有组合,判断两个数的和是否等于目标。如果等于目标,则返回这两个数的索引。 此方法的时间复杂度为O(n^2),空间复杂度为O(1)。 思路2:哈希表 为了优化时间复杂度,可以使用哈希表来存储数组中的元素和对应的索引。遍历数组,对于每个元素nums[i],我们可以通过计算target - nums[i]的,查找哈希表中是否存在这个差。 如果存在,则说明找到了两个数的和等于目标,返回它们的索引。如果不存在,将当前元素nums[i]和它的索引存入哈希表中。 此方法的时间复杂度为O(n),空间复杂度为O(n)。 思路3:双指针 如果数组已经排序,可以使用双指针的方法来求解。假设数组从小到大排序,定义左指针left指向数组的第一个元素,右指针right指向数组的最后一个元素。 如果当前两个指针指向的数的和等于目标,则返回它们的索引。如果和小于目标,则将左指针右移一位,使得和增大;如果和大于目标,则将右指针左移一位,使得和减小。 继续移动指针,直到找到个数的和等于目标或者左指针超过了右指针。 此方法的时间复杂度为O(nlogn),空间复杂度为O(1)。 以上三种方法都可以解决问题,选择合适的方法取决于具体的应用场景和要求。如果数组规模较小并且不需要考虑额外的空间使用,则暴力法是最简单的方法。如果数组较大或者需要优化时间复杂度,则哈希表或双指针方法更合适。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值