Medium
Sum of GCD of Formed Pairs — Python
Full explanation · Time O(nlogr) · Space O(n)
# Time: O(nlogr)
# Space: O(n)
# prefix sum, sort, two pointers
class Solution(object):
def gcdSum(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
def gcd(a, b):
while b:
a, b = b, a%b
return a
mx = 0
prefix = []
for x in nums:
mx = max(mx, x)
prefix.append(gcd(mx, x))
prefix.sort()
result = 0
left, right = 0, len(nums)-1
while left < right:
result += gcd(prefix[left], prefix[right])
left += 1
right -= 1
return result