力扣刷题:三数之和(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,容易超时。在对数组进行排序的基础上,这时候可以考虑先固定一个数字,那么另外两个数字就是这个数字的负数,另外两个数字要比固定的这个数大,一个从左边开始,一共从右边开始,遍历整个数组,找出符合条件的数。


63df19d9fa3807f97dda7ca27a7dd20.jpg

大概思路如上图所示,用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");
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容