Sum of Squares of Special Elements — LeetCode 2778 Python Solution

The problem

You are given a 1-indexed integer array nums of length n. An element nums[i] of nums is called special if i divides n, i.e.

Example

Input

nums = [1,2,3,4]

Output

21

Explanation

There are exactly 3 special elements in nums: nums[1] since 1 divides 4, nums[2] since 2 divides 4, and nums[4] since 4 divides 4.

Python solution

class Solution:
    def sumOfSquares(self, nums: List[int]) -> int:
        n = len(nums)
        return sum(x * x for i, x in enumerate(nums, 1) if n % i == 0)

Complexity

Related problems