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(n3)，严重超时。
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;    }};