변수 범위 잘 보자

import sys
sys.setrecursionlimit(10 ** 6)
input = sys.stdin.readline

N = int(input())
cnt = 0

for A in range(1, 501):
    for B in range(1, A+1): # 주어진 변수 범위 잘보자!!!
        if (B**2 + N) > A**2:
            break
        if (B**2 + N) == A**2:
            cnt+=1

print(cnt)

문제에서 주어진 변수의 범위 잘못봐서, 30분 날려먹었다.
0<=B<=500 주어진 조건인데, 1<=B<=500 인줄 알았다;;
꼭 미리 파악해놓자.