【LeetCode】【算法】15. 三数之和-代码

时间:2024-11-12 17:37:29
class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> list = new ArrayList<>();
        // 数组排序
        Arrays.sort(nums);
        if (nums[0] > 0) return list;
        for (int i = 0; i < nums.length; i++) {
            if (i > 0 && nums[i] == nums[i-1]){
                continue;
            }
            int left = i + 1;
            int right = nums.length - 1;
            
            while (left < right){
                if ((nums[i] + nums[left] + nums[right]) > 0){
                    right--;
                } else if ((nums[i] + nums[left] + nums[right]) < 0){
                    left++;
                } else {
                    // 可以进行存储了,但是要注意去重
                    List<Integer> t = new ArrayList<>();
                    t.add(nums[i]);
                    t.add(nums[left]);
                    t.add(nums[right]);
                    list.add(t);
                    while (right > left && nums[right] == nums[right - 1]) right--;
                    while (right > left && nums[left] == nums[left + 1]) left++;
                    right--;
                    left++;
                }
            }
        }
        return list;
    }
}