Update: Find Pivot Index

Author: openset

problems/find-pivot-index/find_pivot_index.go

@@ -6,12 +6,10 @@ func pivotIndex(nums []int) int {
 		right += v
 	}
 	for i, v := range nums {
-		right -= v
-		if left == right {
+		if right -= v; left == right {
 			return i
-		} else {
-			left += v
 		}
+		left += v
 	}
 	return -1
 }

problems/find-pivot-index/find_pivot_index_test.go

@@ -29,6 +29,10 @@ func TestPivotIndex(t *testing.T) {
 			input:    []int{1, 2, 1},
 			expected: 1,
 		},
+		{
+			input:    []int{},
+			expected: -1,
+		},
 	}
 	for _, tc := range tests {
 		output := pivotIndex(tc.input)