【双指针+简化去重操作】Leetcode 15 三数之和

【双指针+简化操作】Leetcode 15 三数之和

    • 解法1

【双指针+简化去重操作】Leetcode 15 三数之和_第1张图片

解法1

新建一个嵌套列表:List> result = new List<>();
初始化一个ArrayList并直接赋值:ArrayList result = new ArrayList<>(Arrays.asList(1, 2, 3, 4));
向列表中添加元素:result.add()

操作流程

  1. nums[i] 从头到尾遍历数组,nums[left]是nums[i]的下一个元素,nums[right]最初指向数组的最后一个元素
  2. nums数组进行排序!!!
  3. 考察nums[i] + nums[left] + nums[right]的情况。如果和大于0,则right–,如果和小于0,则left++,等于0则添加结果

去重操作(不可以包含重复的三元组)

  • 如果nums[i] 大于0,就可以直接拜拜了 return result
  • 对i进行去重:如果 i>0 && nums[i] == nums[i-1] , continue
  • 对left进行去重:如果nums[left] == nums[left+1]。left++
  • 对right进行去重:如果nums[right] == nums[right - 1]。right–

时间复杂度O(N^2)
空间复杂度O(1)

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        //双指针+去重操作
        List<List<Integer>> result = new ArrayList<>();

        //1【对数组进行排序】
        Arrays.sort(nums);

        // nums[i] 从头到尾遍历数组,nums[left]是nums[i]的下一个元素,nums[right]是最后一个元素
        for(int i = 0; i < nums.length; i++){
            
            // 如果nums[i]大于零,就可以直接拜拜
            if(nums[i] > 0) return result;

            int left = i+1;
            int right = nums.length-1;

            if(i > 0 && nums[i] == nums[i-1]) {
                continue; // 对i进行去重
            }

            // 考察nums[i] + nums[left] + nums[right]的情况
            while(left < right){
                if(nums[i] + nums[left] + nums[right] > 0){
                    right--;
                } 
                else if(nums[i] + nums[left] + nums[right] < 0){
                    left++;
                }
                else{
                    //将结果加入结果集
                    result.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 --;
                }       
            }
            
        }

        return result;
    }
}    

你可能感兴趣的:(Leetcode,leetcode,排序算法,算法,java,数据结构)