diff --git a/arrays_and_strings/pivot_index_array.py b/arrays_and_strings/pivot_index_array.py new file mode 100644 index 0000000..03c0c42 --- /dev/null +++ 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