Leetcode刷题——三数之和

大噶猴，前一段比较忙，刷题日记被耽搁了一段时间，从今天起开始恢复。

今天开始刷leetcode中等难度的算法题了，第一道是三数之和，看下题目要求：

思路：

好久没刷题了手非常生，思路也很枯竭，只想到了暴力循环，找到所有相加等于零的vector之后对每一个vector做一个排序然后判断是否相等，erase掉相同的元素。代码是这样的：

class Solution {
public:
    vector> threeSum(vector& nums) {
        vector> result;
        vector temp;
      
        int len = nums.size();
        int sum = 0;
        for(int m = 0;m < len;m++){
            for(int n = m + 1;n < len;n++){
                for(int o = n + 1;o < len;o++){
                    sum = nums[m] + nums[n] + nums[o];
                    if(sum == 0){
                        temp.push_back(nums[m]);
                        temp.push_back(nums[n]);
                        temp.push_back(nums[o]);
                        result.push_back(temp);
                        temp.clear();
                    }
                    sum = 0;
                }
            }
        }
        int l = result.size();
        for(int x = 0;x < l;x++){
            sort(result[x].begin(),result[x].end());
        }
        for(int y = 0;y < l;y++){
            for(int z = y + 1;z < l;z++){
                if(result[y] == result[z]){
                    result.erase(result.begin() + z);
                    l--;
                    z--;
                }
            }
        }
        return result;
    }
};

可以成功跑通，但是超出了时间限制，非常悲伤。

然后借鉴了别人的思路，可以对输入的数字先做一个排序，对负数部分进行循环，看其后面的两数相加是否能与它抵消，当然如果第一个数就是整数就是无解的输入。在计算后面两数的和的时候我们可以设两个迭代器，一个从这个数后第一个数开始，一个从最后一个数开始，如果加和小了就移动第一个迭代器，如果加和大了就移动第二个迭代器，就可以遍历所有情况了。而且因为数字整个都是排序的，如果有重复数字肯定是相邻的，我们在处理完前一个数之后，如果后一个数与之相等就删掉，这样避免了重复情况的出现。代码如下：

class Solution {
public:
    vector> threeSum(vector& nums) {
        vector> res;
        sort(nums.begin(), nums.end());
        for (int k = 0; k < nums.size(); ++k) {
            if (nums[k] > 0) break;
            if (k > 0 && nums[k] == nums[k - 1]) continue;
            int target = 0 - nums[k];
            int i = k + 1, j = nums.size() - 1;
            while (i < j) {
                if (nums[i] + nums[j] == target) {
                    res.push_back({nums[k], nums[i], nums[j]});
                    while (i < j && nums[i] == nums[i + 1]) ++i;
                    while (i < j && nums[j] == nums[j - 1]) --j;
                    ++i; --j;
                } else if (nums[i] + nums[j] < target) ++i;
                else --j;
            }
        }
        return res;
    }
};

可见刷题是不能断的，感觉自己退步了，sad，还有很多要学习，加油！

我们下期见！