본문 바로가기

Algorithm/Brute Force

(Python) - LeetCode (easy) 1925. Count Square Sum Triples

반응형

https://leetcode.com/problems/count-square-sum-triples/description/

전수조사 문제였습니다.

📕 풀이방법

📔 입력 및 초기화

정답 변수 ans선언 후 0으로 초기화합니다.

📔 풀이과정

$ a^2 + b^2 = c^2 $에서 c = $\sqrt{a^2 + b^2}$ 가 됩니다.

1부터 n까지 이중 for loop를 수행하며 다음을 수행합니다.

를 수행했을 때 c값이 정수면서 n이하라면 ans를 1씩 추가합니다.

📔 정답 출력 | 반환

ans를 반환합니다.


📕 Code

📔 python

class Solution(object):
    def countTriples(self, n):
        ans = 0
        for a in range(1, n+1):
            for b in range(1, n+1):
                c = (a**2 + b**2) ** 0.5
                if c.is_integer() and c <= n:
                    ans += 1
        return ans

*더 나은 내용을 위한 지적, 조언은 언제나 환영합니다.