题解 | #三数之和#TOP54

思路:
1.先排序,不适用暴力破解法
2.使用滑动窗口(两路指针,中间夹),我们先固定窗口位置 i..j..k,i范围可以从0到n-3,j的范围是i+1到k-1,k的范围是j+1到n-1
3.示例,第一轮遍历,固定i为0,j从i+1= 1开始,k从n-1开始,如果三个数相加大于0,那说明后面两数相加和太大,那就要k--,大数变小点,如果小于0,那说明j的数不够大,j++。
4.注意去重

import java.util.*;
public class Solution {
    public ArrayList<ArrayList<Integer>> threeSum(int[] num) {
        
        if(num == null || num.length < 3){
            return new ArrayList<ArrayList<Integer>>();
        }
        //1.排序
        Arrays.sort(num);
        if(num[0] > 0){
            return new ArrayList<ArrayList<Integer>>();
        }
        ArrayList<ArrayList<Integer>> result = new ArrayList<>();
        //1.暴力破解法
//         for(int i = 0;i< num.length - 2;i++){
//             for(int j = i + 1;j<num.length -1;j++){
//                 for(int k = j+ 1; k< num.length;k++){
//                     if(num[i] +num[j] + num[k] == 0){
//                         ArrayList<Integer> ele = new ArrayList<>();
//                         ele.add(num[i]);
//                         ele.add(num[j]);
//                         ele.add(num[k]);
//                         ele.sort((x,y) -> x-y);
//                         if(!result.contains(ele)){
//                             result.add(ele);
//                         }
//                     }
//                 }
//             }
//         }
        //i...j...k
        int n = num.length;
        //2.方法二,两路指针,往中间夹定律。题目中说了不能包含重复的三元组
        for(int i = 0;i< n-2;i++){
            //去重
            if(i > 0 && num[i-1] == num[i]){
                continue;
            }
            int k = n - 1;
            for(int j = i+1; j < k;){
                if(j > i+1 && num[j-1] == num[j]){
                    j++;
                    continue;
                } 
                if(k < n-1 && num[k+1] == num[k]){
                    k --;
                    continue;
                }
                int number = num[i] + num[j] +num[k];
                if(number == 0){
                    ArrayList<Integer> temp = new ArrayList<>();
                    temp.add(num[i]);
                    temp.add(num[j]);
                    temp.add(num[k]);
                    result.add(temp);
                    j++;
                    k --;
                }else if(number > 0){
                    k --;
                }else{
                    j ++;
                }
                
            }
            
        }
        
        
        return result;
    }
}
全部评论

相关推荐

自从我室友在计算机导论课上听说了“刷&nbsp;LeetCode&nbsp;是进入大厂的敲门砖”,整个人就跟走火入魔了一样。他在宿舍门口贴了一张A4纸,上面写着:“正在&nbsp;DP,请勿打扰,否则&nbsp;Time&nbsp;Limit&nbsp;Exceeded。”日记本的扉页被他用黑色水笔加粗描了三遍:“Talk&nbsp;is&nbsp;cheap.&nbsp;Show&nbsp;me&nbsp;the&nbsp;code。”连宿舍聚餐,他都要给我们讲解:“今天的座位安排可以用回溯算法解决,但为了避免栈溢出,我建议用动态规划。来,这是状态转移方程:dp[i][j]&nbsp;代表第&nbsp;i&nbsp;个人坐在第&nbsp;j&nbsp;个位置的最优解。”我让他去楼下取个快递,他不直接去,非要在门口踱步,嘴里念念有词:“这是一个图的遍历问题。从宿舍楼(root)到驿站(target&nbsp;node),我应该用&nbsp;BFS&nbsp;还是&nbsp;DFS?嗯,求最短路径,还是广度优先好。”和同学约好出去开黑,他会提前发消息:“集合点&nbsp;(x,&nbsp;y),我们俩的路径有&nbsp;k&nbsp;个交点,为了最小化时间复杂度,应该在&nbsp;(x/2,&nbsp;y/2)&nbsp;处汇合。”有一次另一个室友低血糖犯了,让他帮忙找颗糖,他居然冷静地分析道:“别急,这是一个查找问题。零食箱是无序数组,暴力查找是&nbsp;O(n)。如果按甜度排序,我就可以用二分查找,时间复杂度降到&nbsp;O(log&nbsp;n)。”他做卫生也要讲究算法效率:“拖地是典型的岛屿问题,要先把连通的污渍区块都清理掉。倒垃圾可以用双指针法,一个指针从左往右,一个从右往左,能最快匹配垃圾分类。”现在我们宿舍的画风已经完全变了,大家不聊游戏和妹子,对话都是这样的:“你&nbsp;Two&nbsp;Sum&nbsp;刷了几遍了?”“别提了,昨天遇到一道&nbsp;Hard&nbsp;题,我连暴力解都想不出来,最后只能看题解。你呢?”“我动态规划还不行,总是找不到最优子结构。今天那道接雨水给我整麻了。”……LeetCode&nbsp;真的害了我室友!!!
老六f:编程嘉豪来了
AI时代还有必要刷lee...
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务