LeetCode 15. 3sum

题意描述

Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note: The solution set must not contain duplicate triplets.

For example, given array S = [-1, 0, 1, 2, -1, -4],
A solution set is:
[
[-1, 0, 1],
[-1, -1, 2]
]

分析

初步想法，分而治之。时间复杂度$O(n^3)$，严重超时。
1. 首先本问题是找到三个数，使得他们的和为0：依次枚举S中的每一个数字，假定它就是使得$a+b+c = 0$中的$a$；
2. 接下来要处理的就是找到两个数，使得他们的和为$-a$：依次枚举S中的每一个数字，假定它就是使得$b+c = -a$的$b$，并且保证$b$与$a$不是同一个数字，即在数组S中的下标相同；
3. 最后判断数组中是否存在一个$c$满足，$c = -a-b$，且保证$a$ 、$b$、$c$中任意两个数的下标都不相同。
注意：输出结果的组合方式不能相同，这里使用$set$去重

实现代码

class Solution {
public:
    vector<vector<int> > threeSum(vector<int>& nums) {
        vector<vector<int> > result;
        set<vector<int> > s;
        for (int z = 0; z < nums.size(); ++z) {
            int twosum = nums[z] * (-1);

            for (int i = 0;i < nums.size(); ++i) {
                vector<int> two; 
                if (i == z) continue;
                int lastNum = twosum - nums[i];
                //bool flag = false;
                for (int k = 0; k < nums.size(); ++k) {
                    if (k == i  || (k == z)) continue;
                    if (lastNum == nums[k]) {
                        two.push_back(nums[i]);
                        two.push_back(lastNum);
                        two.push_back(nums[z]);
                        sort(two.begin(), two.end());
                        s.insert(two);
                         break;
                    }
                }
            }

        }
        for (auto it = s.begin(); it != s.end(); ++it) {
            result.push_back(*it);
        }
        return result;
    }

};