Maximum Total Importance of Roads LeetCode solution in Python

1
class Solution:
2
    def maximumImportance(self, n: int, roads: List[List[int]]) -> int:
3
        Arr = [0] * n  # i-th city has Arr[i] roads
4
        for A, B in roads:
5
            Arr[A] += 1  # Each road increase the road count
6
            Arr[B] += 1
7
        Arr.sort()  # Cities with most road should receive the most score
8
        summ = 0
9
        for i in range(len(Arr)):
10
            summ += Arr[i] * (i + 1)  # Multiply city roads with corresponding score
11

12
        return summ