程序员面试的那些小事程序员架构算法设计模式和编程理论

lintcode 三数之和

2017-01-04  本文已影响145人  yzawyx0220

给出一个有n个整数的数组S,在S中找到三个整数a, b, c,找到所有使得a + b + c = 0的三元组。
注意事项
在三元组(a, b, c),要求a <= b <= c。
结果不能包含重复的三元组。
样例
如S = {-1 0 1 2 -1 -4}, 你需要返回的三元组集合的是:
(-1, 0, 1)
(-1, -1, 2)
题目链接:http://www.lintcode.com/zh-cn/problem/3sum/#

先对数组进行排序,然后设置一个指针,从第三个数开始依次往后遍历,在这个指针指向的数之前再找到两个数,使三个数的和等于target即可。

class Solution {
public:    
    /**
     * @param numbers : Give an array numbers of n integer
     * @return : Find all unique triplets in the array which gives the sum of zero.
     */
    vector<vector<int> > threeSum(vector<int> &nums) {
        // write your code here
        sort(nums.begin(),nums.end());
        vector<vector<int> > res;
        for (int i = 2;i < nums.size();i++) {
            if (i+1 < nums.size() && nums[i] == nums[i+1]) continue;
            for (int j = i-1,k = 0,sum = nums[j] + nums[k],target = 0 - nums[i];k  < j;sum = nums[j] + nums[k]) {
                if (sum < target || k > 0 && nums[k-1] == nums[k]) k++;
                else if (sum > target || j+1 < i && nums[j] == nums[j+1]) j--;
                else res.push_back({nums[k++],nums[j--],nums[i]});
            }
        }
        return res;
    }
};
上一篇下一篇

猜你喜欢

热点阅读