[python]Is my friend cheating?

https://www.codewars.com/kata/5547cc7dcad755e480000004/train/python

Codewars: Train your coding skills

내 코드

# My Code
def removNb(n):
    answer = []
    sum = int((n+1)*n/2)
    # a를 1부터 n 만큼 증가시키면서 b를 찾는다.
    for a in range(1, n+1):
        # b를 수식으로 정리하면 아래와 같다.
        # b = (sum - a - b ) // a
        b = (sum - a) // (a + 1)
        # b가 n보다 작고
        if b <= n:
            # 아래와 같은 조건을 만족하면
            if a * b == sum - a - b:
                # list에 저장한다.
                answer.append((a, b))
    return answer
 
if __name__=='__main__':
    answer = removNb(26)
    print(answer)