[Python] 백준 6131번 완전 제곱수

문제

상근이는 선영이와 함께 게임을 하고 있다. 먼저, 상근이는 두 양의 정수 A와 B를 고른다. (1 ≤ B ≤ A ≤ 500) 

그 다음, 선영이는 상근이가 고른 수를 맞춰야 한다.
상근이는 선영이에게 다음과 같은 힌트를 주었다.
A의 제곱은 B의 제곱보다 N만큼 커 (1 ≤ N ≤ 1,000)
위의 힌트 조건을 만족하는 A와 B 쌍의 개수를 구하는 프로그램을 작성하시오.

 

https://www.acmicpc.net/problem/6131

6131번: 완전 제곱수

입력

첫째 줄에 N이 주어진다.

출력

상근이의 힌트 조건을 만족하는 (A,B) 쌍의 개수를 출력한다. 

예제

# input
15
# output
2

문제 풀이

N = int(input())
answer = 0

for A in range(1, 501):
    for B in range(1, A+1):
        if A**2 == B**2 + N:
            answer += 1
            
print(answer)

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,

혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.