leetcode 3sum算法

题目出处

https://leetcode.com/problems/3sum/

给定一个数组，查找所有和等于0的三个数。

如： 给定的数组为：[-1, 0, 1, 2, -1, -4]

     返回的结果是:[ [-1, -1, 2], [-1, 0, 1]]

分析

可以采用基本算法来解决：

先对数组进行排序

定义三个指针： i, j, k

i: 从排序后的数组开始移动。

k: 从数组的最后往前移动。

j: 在[i, k]之间移动

如果i, j, k对应的数之和为sum;

如果sum== 0, 得到一个解, 同时移动j, k。

如果sum > 0: 移动k

如果sum < 0: 移动j

如果 j > k: 则移动i, 并重新定义 k, j.

同时对算法作两点优化来减少比较的次数

     k :没必要每次都从最后元素开始移动，所以需要记录k的最后位置。

     i: 没必要移动最后， 要以i > 0 为终点。

解法二

还有一种比较快的办法，就是利用hash表。估算了一下，算法复杂度都在o(n*n), 差不多， 就没实现了。

代码

vector<vector<int>> threeSum(vector<int>& nums) {         if(nums.size() < 3) return vector<vector<int>>();        sort(nums.begin(), nums.end());        if(nums.front() > 0 || nums.back() < 0) return vector<vector<int>>();                int last = nums.size()-1;        vector<vector<int>> ret;        for(int i = 0; nums[i] <= 0; i++){            if(i > 0 && nums[i] == nums[i-1]) continue;             int j = i + 1;            int k = last;             while(j < k){                if(j > i+1 && nums[j] == nums[j-1]) {j += 1;continue;}                 int num = nums[i] + nums[j] + nums[k];                if(num < 0){j += 1;}                else if(num > 0){                     k -= 1;                     if(j == i + 1) last = k;                 }                else{                    ret.push_back({nums[i], nums[j], nums[k]});                    j += 1;                    k += 1;                }            }        }        return ret;     }