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; 
    }