今天是 Kevin 的算法之路的第 21 天。为大家讲解 LeetCode 第 15 题,正好借昨天讲解的 todo链接 在分享下这道 三数之和。可能有点难度,希望大家能动动聪明的小脑瓜,一起坚持下去~
每日一笑
如果比我优秀的人不会努力,他们凭什么比我优秀?
题目描述
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
双指针
这道题是典型的双指针应用,双指针和此题的面试出镜率都很高,建议掌握!
这道题的算法思路如下:
- 特殊判断,对于数组长度 n,如果数组为 null 或者数组长度小于 3,直接返回 []。
- 对数组进行排序(排序是必须的,否则无法判断及指针移动)。
- 遍历排序后数组:
- 若 nums[i]>0:因为已经排序好,所以后面不可能有三个数加和等于 0,直接返回结果。
- 对于重复元素:跳过,避免出现重复解
- 令左指针 L=i+1,右指针 R=n-1,当 L<R 时,执行循环:
- 当 nums[i]+nums[L]+nums[R]==0,满足要求将元素添加到结果集中,执行循环,判断左界和右界是否和下一位置重复,去除重复解。并同时将 L,R 移到下一位置,寻找新的解
- 若和大于 0,说明 nums[R] 太大,R 左移
- 若和小于 0,说明 nums[L] 太小,L 右移
代码实现
//go
func threeSum(nums []int) [][]int {
var result [][]int // 记录返回结果
length := len(nums) // 获取数组长度
if nums == nil || length < 3 {
return result
}
sort.Ints(nums) // 排序
for i := 0; i < length; i++ {
// 如果当前数字大于0,则三数之和一定大于0,所以结束循环
if nums[i] > 0 {
break
}
// 去重,下标 i 的数
if i > 0 && nums[i] == nums[i-1] {
continue
}
l := i + 1
r := length - 1
for {
if l >= r {
break
}
sum := nums[i] + nums[l] + nums[r]
if sum > 0 {
r --
continue
}
if sum < 0 {
l ++
continue
}
if sum == 0 {
result = append(result, []int{nums[i], nums[l], nums[r]})
// 去重,下标 l
for l < r && nums[l] == nums[l+1] {
l++
}
// 去重,下标 r
for l < r && nums[r] == nums[r-1] {
r--
}
l++
r--
}
}
}
return result
}
//java
class Solution {
public static List<List<Integer>> threeSum(int[] nums) {
List<List<Integer>> ans = new ArrayList();
int len = nums.length;
if(nums == null || len < 3) return ans;
Arrays.sort(nums); // 排序
for (int i = 0; i < len ; i++) {
if(nums[i] > 0) break; // 如果当前数字大于0,则三数之和一定大于0,所以结束循环
if(i > 0 && nums[i] == nums[i-1]) continue; // 去重
int L = i+1;
int R = len-1;
while(L < R){
int sum = nums[i] + nums[L] + nums[R];
if(sum == 0){
ans.add(Arrays.asList(nums[i],nums[L],nums[R]));
while (L<R && nums[L] == nums[L+1]) L++; // 去重
while (L<R && nums[R] == nums[R-1]) R--; // 去重
L++;
R--;
}
else if (sum < 0) L++;
else if (sum > 0) R--;
}
}
return ans;
}
}
郑重声明:
所展示代码已通过 LeetCode 运行通过,请放心食用~
在唠唠嗑
很多人都想养成好习惯,但大多数人却是三分钟热度。为了我们能一起坚持下去,决定制定如下计划(福利)
一起学算法,打卡领红包!
参与方式:
关注我的微信公众号「Kevin的学堂」
还可「加群」与众多小伙伴
一起坚持,一起学习,一起更优秀!
打卡规则为:
「留言」“打卡XXX天” ➕「分享」到朋友圈
奖励:
连续打卡
21
天,联系本人获取6.6
元红包一个!连续打卡
52
天,联系本人获取16.6
元红包一个!连续打卡
100
天,联系本人获取66.6
元红包一个!
个人精力不够,内容主要发布在公众号平台,其他平台可能更新不及时,还望见谅~