给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请你返回所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
解释:
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意,输出的顺序和三元组的顺序并不重要。
示例 2:
输入:nums = [0,1,1]
输出:[]
解释:唯一可能的三元组和不为 0 。
示例 3:
输入:nums = [0,0,0]
输出:[[0,0,0]]
解释:唯一可能的三元组和为 0 。
以下是 Objective-C 代码,用于查找三数之和为 0 的唯一三元组:
+ (NSArray<NSArray<NSNumber *> *> *)hy_threeSum:(NSArray<NSNumber *> *)nums {
NSMutableArray<NSArray<NSNumber *> *> *result = [NSMutableArray array];
if (nums.count < 3) {
return result;
}
// 对数组进行排序
NSArray<NSNumber *> *sortedNums = [nums sortedArrayUsingSelector:@selector(compare:)];
for (NSInteger i = 0; i < sortedNums.count - 2; i++) {
// 如果当前数字与上一个数字相同,则跳过以避免重复结果
if (i > 0 && [sortedNums[i] isEqualToNumber:sortedNums[i - 1]]) {
continue;
}
NSInteger left = i + 1;
NSInteger right = sortedNums.count - 1;
while (left < right) {
NSInteger sum = sortedNums[i].integerValue + sortedNums[left].integerValue + sortedNums[right].integerValue;
if (sum == 0) {
[result addObject:@[sortedNums[i], sortedNums[left], sortedNums[right]]];
// 跳过重复的数字
while (left < right && [sortedNums[left] isEqualToNumber:sortedNums[left + 1]]) {
left++;
}
while (left < right && [sortedNums[right] isEqualToNumber:sortedNums[right - 1]]) {
right--;
}
left++;
right--;
} else if (sum < 0) {
left++;
} else {
right--;
}
}
}
return result;
}
该算法首先对数组进行排序,然后遍历排序后的数组并选择一个数字作为三元组中的第一个数字。接下来,使用双指针算法在剩余的数字中查找两个数字,使它们的和等于目标值(即当前数字的相反数)。
由于答案不能包含重复的三元组,因此我们需要跳过具有相同值的数字。这可以通过检查左右指针所在位置的前一个或后一个数字来实现。