-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.go
64 lines (46 loc) · 1.38 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
package main
import "fmt"
// 给定一个整数类型的数组 nums,请编写一个能够返回数组“中心索引”的方法。
// 我们是这样定义数组中心索引的:数组中心索引的左侧所有元素相加的和等于右侧所有元素相加的和。
// 如果数组不存在中心索引,那么我们应该返回 -1。如果数组有多个中心索引,那么我们应该返回最靠近左边的那一个。
// 示例 1:
// 输入:
// nums = [1, 7, 3, 6, 5, 6]
// 输出: 3
// 解释:
// 索引3 (nums[3] = 6) 的左侧数之和(1 + 7 + 3 = 11),与右侧数之和(5 + 6 = 11)相等。
// 同时, 3 也是第一个符合要求的中心索引。
// 示例 2:
// 输入:
// nums = [1, 2, 3]
// 输出: -1
// 解释:
// 数组中不存在满足此条件的中心索引。
// 说明:
// nums 的长度范围为 [0, 10000]。
// 任何一个 nums[i] 将会是一个范围在 [-1000, 1000]的整数。
func main() {
// nums := []int{1, 7, 3, 6, 5, 6}
// nums := []int{1, 2, 7, 3}
nums := []int{-1, -1, -1, 0, 1, 1}
ri := pivotIndex(nums)
fmt.Println(ri)
}
func pivotIndex(nums []int) int {
nl := len(nums)
var sum1, sum2 int
for index := 0; index < nl; index++ {
sum1 = 0
sum2 = 0
for i := 0; i < index; i++ {
sum1 = sum1 + nums[i]
}
for j := index + 1; j < nl; j++ {
sum2 = sum2 + nums[j]
}
if sum1 == sum2 {
return index
}
}
return -1
}