LeetCode. 15 - 三数之和

三数之和题目链接

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

注意：答案中不可以包含重复的三元组。

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

解题思路：

1、先将一维数组升序排序，第 1 、2、3 三个数也按照升序选取，但出现符合题目要求的组合出现则加入要返回的二维数组中。

2、在新的组合出现时判断是否与前一个组合相同，若是则放弃该组合。

3、初始化申请二维数组大小为 64 若不够则扩充 2 倍大小。

4、优化代码，防止超时。第 1 个数为正数时即可结束循环，将三重循环优化成双重循环。

int compare(const void  *a, const void *b)
{
	return (*(int*)a - *(int*)b);
}
int** threeSum(int* nums, int numsSize, int* returnSize)
{
	qsort(nums, numsSize, sizeof(int), compare);
	int flag = numsSize, size = 64;
	for (int i = 0; i < numsSize; i++)
		if (nums[i] > 0)
		{
			flag = i; break;
		}
	if (flag == 0 || flag > numsSize - 2)
		flag = numsSize - 2;
	int **arrys = (int **)malloc(size * sizeof(int *));
	*returnSize = 0;
	for (int i = 0; i < flag; i++)
	{
		int  start = i + 1, end = numsSize - 1;
		if (i > 0 && nums[i] == nums[i - 1])
			continue;
		while (start < end)
		{
			int sum = nums[i] + nums[start] + nums[end];
			if (sum== 0)
			{
				if (start > i + 1 && nums[start] == nums[start - 1])
				{
					start++; continue;
				}
				arrys[*returnSize] = (int*)malloc(3 * sizeof(int));
				arrys[*returnSize][0] = nums[i]; arrys[*returnSize][1] = nums[start]; arrys[*returnSize][2] = nums[end];
				//printf("%d %d %d\n", arrys[*returnSize][0], arrys[*returnSize][1], arrys[*returnSize][2]);
				(*returnSize)++;
				if (size == *returnSize)
				{
					size <<= 1;
					arrys = (int **)realloc(arrys, sizeof(int *) * size);
				}
				start++; end--;//因为是从小到大排序，start 增大 则第 3 个数需要减小
				continue;
			}
			else if (sum > 0)
				end--;
			else
			{
				start++;// end = numsSize - 1;
			}
		}
	}
	return arrys;
}