三数之和.-CSDN博客

本文链接：https://blog.csdn.net/qq_43135259/article/details/123995902

思路

如果nums为null或者长度小于3;返回null
对nums进行排序;定义返回结果res
循环nums
- 如果当前元素大于0,后面的数越来越大,不会有符合条件的三个数,结束循环
- 如果当前元素不是第一个并且与上一个重复跳过当前循环
- 定义left和right两个指针;left为当前下标+1;right为最后一个下标
- 循环:结束条件left不小于right
  - 求出当前元素和left和right的和sum
  - 如果sum等于0,记录当前3个元素到res里面;并且移动left和right;
  - 再移动的过程中,需要判断left/right的下一个元素是否与当前元素相等;如果相等要继续移动,直至不等为止
  - 如果sum小于0;移动left指针
  - 如果sum大于0移动right指针
将res返回即可

java

    public List<List<Integer>> threeSum(int[] nums) {
        List<List<Integer>> res = new ArrayList<>();
        if (nums == null || nums.length < 3) {
            return res;
        }
        Arrays.sort(nums);
        int length = nums.length;
        for (int i = 0; i < length; i++) {
            if (nums[i] > 0) {
                break;
            }
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }
            int left = i + 1;
            int right = length - 1;
            while (left < right) {
                int sum = nums[i] + nums[left] + nums[right];
                if (sum == 0) {
                    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--;
                    continue;
                }
                if (sum < 0) {
                    left++;
                    continue;
                }
                right--;
            }
        }
        return res;
    }

go

import "sort"

func threeSum(nums []int) [][]int {
	res := make([][]int, 0)
	if nums == nil || len(nums) < 3 {
		return res
	}
	sort.Ints(nums)
	length := len(nums)
	for i, v := range nums {
		if v > 0 {
			break
		}
		if i > 0 && v == nums[i-1] {
			continue
		}
		left, right := i+1, length-1
		for left < right {
			sum := v + nums[left] + nums[right]
			if sum == 0 {
				res = append(res, []int{v, nums[left], nums[right]})
				for left < right && nums[left] == nums[left+1] {
					left++
				}
				for left < right && nums[right] == nums[right-1] {
					right--
				}
				left++
				right--
				continue
			}
			if sum < 0 {
				left++
				continue
			}
			right--
		}
	}
	return res
}