题目
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?
请你找出所有和为 0 且不重复的三元组。注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例 2:
输入:nums = []
输出:[]
示例 3:
输入:nums = [0]
输出:[]
提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
《中级算法》的第一道题,就难倒我了
说实话,从评论区看了思路,觉得自己可以了
写出的代码只能计算出来普通数组
像[],[0,0,0]这种总会报错
就算是照着评论区大佬的过程+自己的思想,还是会报错
例如
nums == NULL || numsSize < 3 的判断与nums[0] > 0) || (nums[numsSize - 1] < 0不能都写在qsort的下边
又例如while (left < right && nums[++left] == nums[left - 1]);
稍稍改一点也会错
附上大佬的代码链接:https://leetcode-cn.com/problems/3sum/solution/15-san-shu-zhi-he-by-dyzahng-xmz9/
int Cmp_int(const void* p1, const void* p2)
{
return *(int*)p1 - *(int*)p2;
}
int** threeSum(int* nums, int numsSize, int* returnSize, int** returnColumnSizes)
{
if (nums == NULL || numsSize < 3)
{
*returnSize = 0;
return NULL;
}
int** arr = (int**)malloc(sizeof(int*) * numsSize * numsSize);
//分配每一列的空间
*returnColumnSizes = (int*)malloc(sizeof(int) * numsSize * numsSize);
qsort(nums, numsSize, sizeof(int), Cmp_int);
if ((nums[0] > 0) || (nums[numsSize - 1] < 0))
{
*returnSize = 0;
return NULL;
}
int left, right, tmp;
int j = 0;
for (int i = 0; i < numsSize; i++)
{
if (nums[i] > 0)
{
break;
}
if (i > 0 && nums[i] == nums[i - 1])
{
continue;
}
left = i + 1;
right = numsSize - 1;
while (left < right)
{
if (nums[i] + nums[left] + nums[right] == 0)
{
//分配每一行具体空间
arr[j] = (int*)malloc(sizeof(int) * 3);
arr[j][0] = nums[i];
arr[j][1] = nums[left];
arr[j][2] = nums[right];
//每一行的列数
(*returnColumnSizes)[j] = 3;
j++;
while (left < right && nums[++left] == nums[left - 1]);
while (left < right && nums[--right] == nums[right + 1]);
}
else if (nums[i] + nums[left] + nums[right] < 0)
{
left++;
}
else
{
right--;
}
}
}
*returnSize = j;
return arr;
}