[파이썬/python] 백준 6131번 완전 제곱수

 

 

 

 

문제

 

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

상근이는 선영이에게 다음과 같은 힌트를 주었다.

A의 제곱은 B의 제곱보다 N만큼 커 (1 ≤ N ≤ 1,000)

위의 힌트 조건을 만족하는 A와 B 쌍의 개수를 구하는 프로그램을 작성하시오.

 

 

 

 

 

 

 

 

입력

 

첫째 줄에 N이 주어진다.

ex)

15

 

 

 

 

 

 

 

 

출력

 

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

ex)

2

 

---

---

 

코드

 

import sys

n=int(sys.stdin.readline())

count=0

for i in range(1,n):
    if i**2 - (i-1)**2 > n:
        break
    for j in range(1,i+1):
        if i**2 - j**2 == n:
            count=count+1
            
print(count)