标签:剪枝 整数 eof sort 操作 集合 amp log 位置
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
//最大答案数 #define MAX_ANS_NUM 1000000 int comp(const void* a, const void* b) { return *((int*)a) - *((int*)b); } int** threeSum(int* nums, int numsSize, int* returnSize, int** returnColumnSizes) { int** result = (int**)calloc(MAX_ANS_NUM, sizeof(int*)); *returnColumnSizes = (int*)calloc(MAX_ANS_NUM, sizeof(int)); *returnSize = 0; if (numsSize < 3) return result; //先将入参排序 qsort(nums, numsSize, sizeof(int), comp); //排除极端异常情况: if (nums[0] + nums[1] + nums[2] > 0) return result; // 最小三个数相加都大于0直接无解 if (nums[numsSize - 1] + nums[numsSize - 2] + nums[numsSize - 3] < 0) return result; //最大三个数相加都小于0直接无解 int left, right, anscount; anscount = 0; for (int i = 0; i < numsSize - 2; i++) { if (nums[i] + nums[i + 1] + nums[i + 2]) { if (nums[i] + nums[numsSize - 1] + nums[numsSize -2] < 0) continue; // 剪枝优化无效判断 最小的加上 最大的两个数都小于0继续找下一轮 } if (i > 0 && nums[i] == nums[i - 1]) { continue; // 重复的判断 } left = i + 1; right = numsSize - 1; while (left < right) { if (nums[left] + nums[i] + nums[right] == 0) { result[anscount] = (int*)calloc(3, sizeof(int)); result[anscount][0] = nums[left]; result[anscount][1] = nums[i]; result[anscount][2] = nums[right]; (*returnColumnSizes)[anscount] = 3; anscount++; while(left + 1 < right && nums[left] == nums[left + 1]) { left++; } left++; while (left < right - 1 && nums[right] == nums[right - 1]) { right--; } right--; } else if (nums[left] + nums[i] + nums[right] < 0) { left++; } else if (nums[left] + nums[i] + nums[right] > 0) { right--; } } } *returnSize = anscount; return result; }
标签:剪枝 整数 eof sort 操作 集合 amp log 位置
原文地址:https://www.cnblogs.com/mhq-martin/p/12019500.html