diff --git a/swift/find-pivot-index.swift b/swift/find-pivot-index.swift new file mode 100644 index 0000000..4987c02 --- /dev/null +++ b/swift/find-pivot-index.swift @@ -0,0 +1,18 @@ +class Solution { + func pivotIndex(_ nums: [Int]) -> Int { + var fromLeft = 0 + var fromRight = nums.reduce(0, +) + + for (i, x) in nums.enumerated() { + fromRight -= x + + if fromLeft == fromRight { + return i + } + + fromLeft += x + } + + return -1 + } +}