力扣刷题:三数之和(java实现)

题目:给你一个包含 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]
输出:[]

提示:

  • 0 <= nums.length <= 3000
  • 10^5 <= nums[i] <= 10^5

题目要求找出和为0的三个数字,如果直接使用三个for循环,会导致时间复杂度很高,比如3000的三次方,时间复杂度有10^9,容易超时。在对数组进行排序的基础上,这时候可以考虑先固定一个数字,那么另外两个数字就是这个数字的负数,另外两个数字要比固定的这个数大,一个从左边开始,一共从右边开始,遍历整个数组,找出符合条件的数。请添加图片描述
大概思路如上图所示,用for循环对固定的 那个遍历,用while循环另外两个指针。代码如下:

public List<List<Integer>> threeSum(int[] nums) {
    
    
        Arrays.sort(nums);
        List<List<Integer>> res = new ArrayList<>();
        for (int i = 0; i < nums.length-2; i++) {
    
    
            //重复的数字跳过
            if(i>0 &&nums[i] == nums[i-1]){
    
    
                continue;
            }
            //如果第一个数都大于0了,那后面的都大于0,肯定不符合条件
            if(nums[i] > 0){
    
    
                break;
            }
            int left = i+1;//左指针
            int right = nums.length-1;//右指针
            int target = -nums[i];
            while(left<right){
    
    
                int sum = nums[left] + nums[right];
                if(sum == target){
    
    
                    res.add(Arrays.asList(nums[i],nums[left],nums[right]));
                    while(left<right && nums[left]==nums[left+1]){
    
    
                        left++;
                    }
                    while (left<right && nums[right] == nums[right-1]){
    
    
                        right--;
                    }
                    left++;
                    right--;
                }else if(sum<target){
    
    
                    left++;
                }else {
    
    
                    right--;
                }

            }
        }
        return res;

    }

说明一下Arrays.asList()
该方法是将数组转化成List集合的方法。

List<String> list = Arrays.asList("a","b","c");

Guess you like

Origin blog.csdn.net/will__be/article/details/121195190