Pythonist

Условие:

Пусть d(n) определяется как сумма делителей n (числа меньше n, делящие n нацело).
Если d(a) = b и d(b) = a, где a ≠ b, то a и b называются дружественной парой, а каждое из чисел a и b — дружественным числом.

Например, делителями числа 220 являются 1, 2, 4, 5, 10, 11, 20, 22, 44, 55 и 110, поэтому d(220) = 284. Делители 284 — 1, 2, 4, 71, 142, поэтому d(284) = 220.

Подсчитайте сумму всех дружественных чисел меньше 10000.

Решение:

def compute():
 # Compute sum of proper divisors for each number
 divisorsum = [0] * 10000
 for i in range(1, len(divisorsum)):
  for j in range(i * 2, len(divisorsum), i):
   divisorsum[j] += i
 
 # Find all amicable pairs within range
 ans = 0
 for i in range(1, len(divisorsum)):
  j = divisorsum[i]
  if j != i and j < len(divisorsum) and divisorsum[j] == i:
   ans += i
 return str(ans)


if __name__ == "__main__":
 print(compute())