Four Divisors LeetCode solution in Python

1
import math
2

3

4
class Solution:
5
    def sumFourDivisors(self, nums: List[int]) -> int:
6
        s = 0
7
        for i in nums:
8
            r = i + 1
9
            c = 2
10
            for j in range(2, int(math.sqrt(i)) + 1):
11
                if i % j == 0:
12
                    if i / j == j:
13
                        c += 1
14
                        r += j
15
                    else:
16
                        c += 2
17
                        r += j + int(i / j)
18
            print(c, r)
19
            if c == 4:
20
                s += r
21
        return s