thiagowfx's avatar

Β¬ just serendipity πŸ€ (not just serendipity)

LeetCode #1991: Find the Middle Index in Array

β€’ 47 words β€’ 1 min

LeetCode #1991: Find the Middle Index in Array:

python
from itertools import accumulate

class Solution:
    def findMiddleIndex(self, nums: List[int]) -> int:
        acc = list(accumulate(nums))

        for i, num in enumerate(nums):
            if (acc[i - 1] if i > 0 else 0) == (acc[-1] - acc[i]):
                return i

        return -1