What is the best way to solve Pythagorean Triplets?

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

DSA 섹션쉬움

피타고라스 삼중항

'피타고라스 삼중항' 문제에 대한 자세한 가이드 및 Python 구현.

문제 설명

쉬움

양의 정수 배열 arr을 취하고 배열에 세 개의 정수 a, b, c이 있으면 True를 반환하고 그렇지 않으면 a^2 + b^2 = c^2 또는 False을 반환하는 함수 contains_pythagorean_triplet(arr)을 작성하세요.

제약

•3 <= len(arr) <= 1000
•1 <= arr[i] <= 1000

예

Example 1

Input

arr = [3, 1, 4, 6, 5]

Output

True

Explanation

3^2 + 4^2 = 9 + 16 = 25 = 5^2.

Example 2

Input

arr = [10, 4, 6, 12, 5]

Output

False

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(1)

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

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

최적화된 솔루션 Python 코드

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

def find_triplets_opt(arr):
    arr = [x*x for x in arr]; arr.sort(); n = len(arr)
    for i in range(n-1, 1, -1):
        l, r = 0, i - 1
        while l < r:
            if arr[l] + arr[r] == arr[i]: return True
            if arr[l] + arr[r] < arr[i]: l += 1
            else: r -= 1
    return False

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

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

def find_triplets_brute(arr):
    n = len(arr)
    for i in range(n):
        for j in range(i+1, n):
            for k in range(j+1, n):
                x, y, z = arr[i]**2, arr[j]**2, arr[k]**2
                if x == y+z or y == x+z or z == x+y: return True
    return False