[백준 / 9094] 수학적 호기심 - Python

(a, b)는 정수쌍이므로, for문을 통해서 한번씩 돌려주고, 

간단히 (a2+b2+m)/(ab)에서 /를 %로 바꾸어 조건에 맞으면 count를 추가해 주면 된다.

 

python3로 제출하니 계속 시간초과가 나와서 pypy3로 제출했다.

 

 

 

<전체 코드>

 

import sys

for _ in range(int(sys.stdin.readline())):
    n,m=map(int,sys.stdin.readline().split())
    count=0
    for a in range(1,n-1):
        for b in range(a+1,n):
            if (a**2+b**2+m)%(a*b)==0:count+=1
    print(count)