此题来自力扣 15.三数之和
目录
题目描述
注意: 不可以包含重复的三元组
给你一个包含 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]
输出:[]
题解思路
排序+双指针(最主要的在于去重和双指针的控制)
- 判断是否为空或者数组长度小于3,直接返回空数组
- 对数组进行升序排序
- 循环数组的数据,当数组的开始数据($nums[$i])到达大于0时,那么直接返回结果,因为升序排序之后,后面的数据只会更大,所以相加不会为0.
- 对于循环数组的元素,判断重复之后直接跳过,因为重复的元素在前面已经判断过所有的数据了,结果不可以重复,所以直接跳过.
- 左指针$L = i+1, 右指针$R = count($nums)-1. 当$L < $R时,循环
- 当$nums[$i]+$nums[$L]+$nums[$R] =0时,判断循环,判断$nums[$L]和$nums[$R]的下一位是否和当前的值重复,如果重复,跳过,直到将$L和$R找到下一个位置
- 当$nums[$i]+$nums[$L]+$nums[$R] >0时, 说明$R的值太大了,将$R的值向前移动
- 当$nums[$i]+$nums[$L]+$nums[$R] <0时, 说明$L的值太小了,将$L的值向后移动
代码
<?php
/**
* @param Integer[] $nums
* @return Integer[][]
*/
function threeSum($nums)
{
$n = count($nums);
$res = [];
if (empty($nums) || $n < 3) return $res;
sort($nums);
for ($i = 0; $i < $n; $i++) {
if ($nums[$i] > 0) return $res;
if ($i > 0 && $nums[$i] == $nums[$i - 1]) continue;
$l = $i + 1;
$r = $n - 1;
while ($l < $r) {
if ($nums[$i] + $nums[$l] + $nums[$r] == 0) {
$res[] = [$nums[$i], $nums[$l], $nums[$r]];
while ($nums[$l] == $nums[$l + 1] && $l < $n - 1) {
$l = $l + 1;
}
while ($nums[$r] == $nums[$r - 1] && $l < $n - 1) {
$r = $r - 1;
}
$l = $l + 1;
$r = $r - 1;
} else if ($nums[$i] + $nums[$l] + $nums[$r] < 0) {
$l = $l + 1;
} else {
$r = $r - 1;
}
}
}
return $res;
}
$nums = [-1, 0, 1, 2, -1, -4];
$result = threeSum($nums);
print_r($result);