1. 题目
https://leetcode.cn/problems/3sum/
2. 解法
思路是这样的:
- 首先,对数组进行排序,这样可以方便地跳过重复的元素,也可以利用双指针法来寻找第二个数和第三个数。
- 然后,遍历数组,固定第一个数,如果第一个数大于零,那么后面的数都大于零,不可能有解,所以可以直接结束循环。
- 接着,使用双指针法,从左右两端向中间寻找第二个数和第三个数,如果三数之和为零,就找到一个解,并跳过重复的元素;如果三数之和小于零,就说明第二个数太小,需要增大;如果三数之和大于零,就说明第三个数太大,需要减小。
- 最后,返回结果列表。
第一元素固定+后两个元素采用双指针
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 | public List<List<Integer>> threeSum( int [] nums) { //先对数组进行排序 Arrays.sort(nums); //创建一个结果列表 List<List<Integer>> result = new ArrayList<>(); //遍历数组,从第一个数开始,到倒数第三个数结束 for ( int i = 0 ; i < nums.length - 2 ; i++) { <strong> //如果当前数和前一个数相同,就跳过,避免重复 if (i > 0 && nums[i] == nums[i - 1 ]) { continue ; }</strong> //定义两个指针,分别指向当前数后面的第一个数和最后一个数 int left = i + 1 ; int right = nums.length - 1 ; //当左指针小于右指针时,循环查找 while (left < right) { //计算当前三个数的和 int sum = nums[i] + nums[left] + nums[right]; //如果和为0,就找到了一个解 if (sum == 0 ) { //把这三个数加入到结果列表中 result.add(Arrays.asList(nums[i], nums[left], nums[right])); <strong> //左指针右移,跳过相同的数,避免重复 while (left < right && nums[left] == nums[left + 1 ]) { left++; } //右指针左移,跳过相同的数,避免重复 while (left < right && nums[right] == nums[right - 1 ]) { right--; }</strong> //左指针和右指针都移动一位,继续查找 left++; right--; } else if (sum > 0 ) { //如果和大于0,说明右边的数太大了,需要减小,所以右指针左移 right--; } else { //如果和小于0,说明左边的数太小了,需要增大,所以左指针右移 left++; } } } //返回结果列表 return result; } |
其中以下三处 处理重复 非常重要
1 2 3 4 | //如果当前数和前一个数相同,就跳过,避免重复 if (i > 0 && nums[i] == nums[i - 1 ]) { continue ; } |
1 2 3 4 5 6 7 8 | //左指针右移,跳过相同的数,避免重复 while (left < right && nums[left] == nums[left + 1 ]) { left++; } //右指针左移,跳过相同的数,避免重复 while (left < right && nums[right] == nums[right - 1 ]) { right--; } |
3. 总结
分类:
算法
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享4款.NET开源、免费、实用的商城系统
· 全程不用写代码,我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了
· 上周热点回顾(2.24-3.2)