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
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值