leetcode解题报告15. 3Sum

leetcode解题报告15. 3Sum

题目地址

难度是medium

题目描述

输入一个整数数组，从中选择３个数，使得这三个数之和为０，这三个数称为一个三元组。现在要求所有满足要求的三元组。并且要求三元组之间不重复。（其中三元组里的元素是没有顺序的，即(1,2,3)和(3,2,1)是同一个三元组）

我的思路

首先暴力的办法是遍历所有三元组，找出满足要求的即可。但是一方面复杂度太高了$O(n^3)$的复杂度。另一方面，如何进行三元组的去重呢？
对于第一个问题，可以考虑利用哈希表技巧，使得找一个数的复杂度从$O(n)$变成$O(\log n)$，从而使得复杂度从$O(n^3)$变成$O(n^2\log n)$
对于第二问题，想了好久，尝试了利用stl里的set来去重，但是不太会用，比较麻烦，放弃了。后来想到，三元组的比较麻烦在于顺序问题，可以先要求，三元组里的元素是升序的，那可以先确定第一个位置的元素，然后再确定第二个位置的元素，最后确定第三个位置的元素。如是得到的三元组是唯一的。具体操作是先对数组排序，然后第一个for循环确定第一个位置的元素，然后第二个for循环确定第二个位置的元素。要求是大于第一个位置元素。如是就可以判断第三个元素，要求在数组里，并大于等于第二个元素(如果等于，要保证原来数组里有多个该元素)，且三者和为零。

我的代码

#include <algorithm>
#include <map>

using namespace std;
class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> result;
        //处理不满足要求的情况
        if (nums.size() < 3) {
            return result;
        }
        //要排序，避免改变输入，新建一个数组
        vector<int> t_nums(nums.begin(), nums.end());
        sort(t_nums.begin(), t_nums.end());
        //建立哈希表，加速在一个数组里找一个数
        map<int, int> my_map;
        for (int i = 0; i < t_nums.size(); i++) {
            if (my_map.count(t_nums[i]) > 0) {
                my_map[t_nums[i]] += 1;
            } else {
                my_map[t_nums[i]] = 1;
            }
        }


        int pre = t_nums[0]+1;//初始化成和第一个数不一样
        for (int i = 0; i < t_nums.size()-1; i++) {
      　　//避免重复
            if (t_nums[i] == pre) {
                continue;
            }
            pre = t_nums[i];
            my_map[t_nums[i]] -= 1;
            int pre_2 = t_nums[i+1] + 1;
            for (int j = i+1; j < t_nums.size(); j++) {
                if (t_nums[j] == pre_2) {
                    continue;
                }
                pre_2 = t_nums[j];
                my_map[t_nums[j]] -= 1;
                int target = -(t_nums[i]+t_nums[j]);
                if (target >= t_nums[j] && my_map.count(target) > 0 && my_map[target] > 0) {
                    vector<int> temp;
                    temp.push_back(t_nums[i]);
                    temp.push_back(t_nums[j]);
                    temp.push_back(target);
                    result.push_back(temp);
                }
                my_map[t_nums[j]] += 1;
            }
            my_map[t_nums[i]] += 1;
        }
        return result;
    }
};

阅读官方题解

没有官方题解。网上有个一个$O(n^2)$的思路，大体是先确定第一个位置元素。然后从数组的头尾开始分别找第二个位置元素和第三个位置元素。

思想核心总结

注意思考问题，不要过于依赖stl