[Leetcode] 2913. Subarrays Distinct Element Sum of Squares I

https://leetcode.com/problems/subarrays-distinct-element-sum-of-squares-i

Subarrays Distinct Element Sum of Squares I - LeetCode

class Solution:
    def sumCounts(self, nums: List[int]) -> int:
        count = 0
        subarrays = []

        for i in range(1, len(nums)+1):
            for j in range(len(nums)-i+1):
                subarrays.append(nums[j:i+j])
        
        for sub in subarrays:
            sub = list(set(sub))
            count += len(sub) * len(sub)

        return count