15. 三数之和 @Python @Go

原题

https://leetcode.cn/problems/3sum/description/

思路

双指针

复杂度

时间：O(n * n)
空间：O(n * n)

Python代码

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        # 升序排列
        nums.sort()
        ans = []
        for i in range(len(nums)-2):
            if nums[i] > 0:
                break 
            # 对nums[i]去重
            if i > 0 and nums[i] == nums[i-1]:
                continue
            j = i + 1 
            k = len(nums) - 1 
            while j < k:
                if nums[i] + nums[j] + nums[k] == 0:
                    ans.append([nums[i], nums[j], nums[k]])
                    # 对nums[j]去重 
                    while j < k and nums[j] == nums[j+1]:
                        j += 1 
                    # 对nums[k]去重
                    while j < k and nums[k] == nums[k-1]:
                        k -= 1 
                    j += 1 
                    k -= 1 
                elif nums[i] + nums[j] + nums[k] > 0:
                    k -= 1 
                else:
                    j += 1               

        return ans   

Go代码

func threeSum(nums []int) [][]int {
	// 升序排列
	sort.Ints(nums)
	var ans [][]int
	for i, _ := range nums {
		if nums[i] > 0 {
			break
		}
		// 对nums[i]去重
		if i > 0 && nums[i] == nums[i-1] {
			continue
		}
		j, k := i+1, len(nums)-1
		for j < k {
			if nums[i]+nums[j]+nums[k] == 0 {
				ans = append(ans, []int{nums[i], nums[j], nums[k]})
				// 对nums[j]去重
				for j < k && nums[j] == nums[j+1] {
					j++
				}
				// 对nums[k]去重
				for j < k && nums[k] == nums[k-1] {
					k--
				}
				j++
				k--
			} else if nums[i]+nums[j]+nums[k] > 0 {
				k--
			} else {
				j++
			}
		}
	}
	return ans
}