???双指针中等 leetcode15. 三数之和

排序+双指针
排序,方便区分正负数。三个数和为0必定有一个是小于等于0。
设三个变量,i在小于等于0的区间里遍历,j初始位置是i+1,k初始位置是nums.length-1
i在小于等于0的区间里遍历,每在一个位置,都让j,k遍历数组直到j不小于k.i在遍历时还要注意去重。
当三个数的和小于0,j++
当三个数的和大于0,k–
当三个数的和等于0,保存这三个数,去重

import java.util.ArrayList;
import java.util.Arrays;
class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        if(nums.length < 3){
            return res;
        }
        Arrays.sort(nums);
        for(int i = 0; i < nums.length-2; i++){
            if(nums[i] > 0){
                break;
            }
            if(i > 0 && nums[i] == nums[i-1]){//去重
                continue;
            }
            int j = i + 1;
            int k = nums.length - 1;
            while(j < k){
                int sum = nums[i] + nums[j] + nums[k];
                if(sum < 0){
                    j++;
                }else if(sum > 0){
                    k--;
                }else{
                    List<Integer> li = new ArrayList<>();
                    li.add(nums[i]);
                    li.add(nums[j]);
                    li.add(nums[k]);
                    res.add(li);
                    while(j < k && nums[k-1] == nums[k]){//去重
                        k--;
                    }
                    while(j < k && nums[j+1] == nums[j]){//去重
                        j++;
                    }
                    k--;//很重要,否则超时,还没明白为什么超时。
                    j++;
                }
            }
        }
        return res;
    }
}
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客
应支付0元
点击重新获取
扫码支付

支付成功即可阅读