python求亲密数对

Question

键盘输入n，n在2至2000之间，求2至n中的亲密数对。
就是a的（去掉1和自己的）因子和等于b，b的（去掉1和自己）因子和等于a，且a≠b。
如48和75是亲密数对。
48的因子和为2+3+4+6+8+12+16+24=75，
而75的因子和为3+5+15+25=48。
输入
200
输出
48 75
75 48
140 195
195 140

Answer 1

import math

def factor_sum(n):
    factors = [1]
    for i in range(2, int(math.sqrt(n)) + 1):
        if n % i == 0:
            factors.append(i)
            factors.append(n // i)
    return sum(factors)

def find_amicable_pairs(n):
    pairs = []
    for a in range(2, n+1):
        b = factor_sum(a)
        if a != b and factor_sum(b) == a:
            pairs.append((a, b))
    return pairs

n = int(input("请输入一个正整数n（2~2000）："))
pairs = find_amicable_pairs(n)
for pair in pairs:
    print(pair[0], pair[1])