What is the best way to solve 3Sum?

The optimized O(n) approach using Two Pointers patterns is generally preferred.

상위 150개 인터뷰쉬움

3섬

'3Sum' 문제에 대한 자세한 가이드 및 Python 구현입니다.

문제 설명

쉬움

정수 배열 nums이 주어지면 i != j, i != k, j != k 및 nums[i] + nums[j] + nums[k] == 0와 같은 세 쌍의 [nums[i], nums[j], nums[k]]을 모두 반환합니다.

솔루션 세트에는 중복된 삼중항이 포함되어서는 안 됩니다.

threeSum(nums: List[int]) -> List[List[int]] 함수를 작성하세요.

제약

•3 <= len(nums) <= 3000
•-10^5 <= nums[i] <= 10^5

예

Example 1

Input

nums = [-1, 0, 1, 2, -1, -4]

Output

[[-1, -1, 2], [-1, 0, 1]]

Explanation

nums[0] + nums[1] + nums[2] = -1 + 0 + 1 = 0. nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0. nums[0] + nums[3] + nums[4] = -1 + 2 + (-1) = 0. The distinct triplets are [-1,-1,2] and [-1,0,1].

Example 2

Input

nums = [0, 1, 1]

Output

[]

Explanation

No triplet sums to 0.

Example 3

Input

nums = [0, 0, 0]

Output

[[0, 0, 0]]

Explanation

The only possible triplet sums to 0.

Need a Hint?

세트나 힙과 같은 두 포인터 관련 데이터 구조를 사용하는 것을 고려해보세요.

Edge Cases to Watch

빈 입력 구조
단일 요소 입력
큰 수치 범위

해결할 준비가 되셨나요?

Open the problem in PyRun's browser-based Python editor. Your code runs fully offline — no server required.

편집기에서 열기

인터뷰 통찰력 및 변형

두 포인터 접근 방식의 논리를 설명하세요.

null 또는 빈 입력과 같은 극단적인 경우에 대해 논의합니다.

복잡성 분석 분석

무차별 대입 복잡성

Time: O(n)•Space: O(1)

왜 시간인가?: Directly evaluates all possibilities.

왜 우주인가?: Uses standard local memory.

최적화된 복잡성

Time: O(n)•Space: O(n)

왜 시간인가?: Optimized paths reduce total operations.

왜 우주인가?: May trade memory for speed.

최적화된 솔루션 Python 코드

컴파일러에서 시도해 보세요

def three_sum_opt(nums):
    res = []
    nums.sort()

    for i, a in enumerate(nums):
        if i > 0 and a == nums[i - 1]:
            continue

        l, r = i + 1, len(nums) - 1
        while l < r:
            threeSum = a + nums[l] + nums[r]
            if threeSum > 0:
                r -= 1
            elif threeSum < 0:
                l += 1
            else:
                res.append([a, nums[l], nums[r]])
                l += 1
                while l < r and nums[l] == nums[l - 1]:
                    l += 1
    return res

무차별 대입 코드(스포일러 보호)

컴파일러에서 시도해 보세요

def three_sum_brute(nums):
    res = set()
    nums.sort()
    n = len(nums)
    for i in range(n):
        for j in range(i + 1, n):
            for k in range(j + 1, n):
                if nums[i] + nums[j] + nums[k] == 0:
                    res.add((nums[i], nums[j], nums[k]))
    return [list(t) for t in res]