题目描述
给你一个包含 n n n 个整数的数组 nums,判断 nums 中是否存在三个元素 a , b , c a,b,c a,b,c ,使得 a + b + c = 0 a + b + c = 0 a+b+c=0?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
- 示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[-1,-1,2],[-1,0,1]] - 示例 2:
输入:nums = []
输出:[] - 示例 3:
输入:nums = [0]
输出:[]
题解
一道经典的算法题。思路,首先考虑暴力解法,三层循环,复杂度
O
(
N
3
)
O(N^3)
O(N3),显然复杂度太高,并且还需要剔除掉重复元素。
为了让数组中不要出现重复元素,我们可以先对数组进行排序,复杂度
O
(
N
l
o
g
N
)
O(NlogN)
O(NlogN)(使用快速排序)。这样我们保证了我们的三个数
a
<
b
<
c
a<b<c
a<b<c。对于数组中一个固定的
a
a
a,我们只需要找到后面满足
b
+
c
=
−
a
b+c=-a
b+c=−a的
b
b
b和
c
c
c,由于已排序
(
a
<
b
<
c
)
(a<b<c)
(a<b<c),对于
a
a
a之后一个固定的
b
b
b,我们从尾部进行迭代,直到找到一个
c
c
c使得
a
+
b
+
c
≤
0
a+b+c\leq0
a+b+c≤0,结束循环,判断
a
,
b
,
c
a,b,c
a,b,c是否符合相加为0,若符合,则添加当前三个数
注意:由于我们已经排好序了,所以我们可以通过判断当前元素是否和前一个元素相同而跳过相同元素。以下为代码:
/**
* Return an array of arrays of size *returnSize.
* The sizes of the arrays are returned as *returnColumnSizes array.
* Note: Both returned array and *columnSizes array must be malloced, assume caller calls free().
*/
int CompareByIncrease(const void* a, const void* b)
{
return *(int*)a - *(int*)b; // 快排构造递增序列
}
int **threeSum(int *nums, int numsSize, int *returnSize, int **returnColumnSizes)
{
*returnSize = 0;
if (nums == NULL || numsSize < 3) return NULL;
qsort(nums, numsSize, sizeof(int), CompareByIncrease);
int ** ans = (int**)malloc(sizeof(int*) * numsSize * numsSize); // 二维数组返回值
*returnColumnSizes = (int*)malloc(sizeof(int) * numsSize * numsSize); // 记录列数的数组
for (int i = 0; i < numsSize - 2 && nums[i] <= 0; i++) // 碰到正数时停止,后面都是正数不存在符合题目条件的
{
if (i == 0 || nums[i] != nums[i - 1]) // 去除重复元素
{
int j = i, k = numsSize - 1;
while (++j < k)
if ((j == i + 1) || nums[j] != nums[j - 1])
{
while (k > j + 1 && nums[i] + nums[j] + nums[k] > 0)
k--;
if (nums[i] + nums[j] + nums[k] == 0)
{
ans[*returnSize] = (int*)malloc(sizeof(int) * 3);
(*returnColumnSizes)[*returnSize] = 3;
ans[*returnSize][0] = nums[i];
ans[*returnSize][1] = nums[j];
ans[*returnSize][2] = nums[k];
(*returnSize)++;
}
}
}
}
return ans;
}