Create pivot_index_array.py

2025-04-29 20:26:07 -04:00 · 2023-07-31 16:32:41 -07:00 · 2023-07-31 16:32:41 -07:00 · b467dca561
commit b467dca561
parent 0963a43bcc
1 changed files with 19 additions and 0 deletions
--- a/arrays_and_strings/pivot_index_array.py
+++ b/arrays_and_strings/pivot_index_array.py
@ -0,0 +1,19 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+# author: bt3gl
+
+
+def pivotIndex(nums):
+
+        s = sum(nums)
+
+        left_sum = 0
+
+        for i, x in enumerate(nums):
+
+            if left_sum == (s - left_sum - x):
+                return i
+            
+            left_sum += x
+        
+        return -1