力扣724. 寻找数组的中心索引

原创于 2021-01-28 14:56:08 发布 · 114 阅读

0 ·

CC 4.0 BY-SA版权

go 专栏收录该内容

35 篇文章

订阅专栏

这篇文章介绍了如何使用pivotIndex函数高效地在整数数组中找到使得前半部分和等于后半部分和加当前元素的索引。通过循环和条件判断，函数在O(n)时间内求解。示例代码展示了nums数组中符合条件的索引计算过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

func pivotIndex(nums []int) int {
	total := 0
	for _, v := range nums {
		total += v
	}
	sum := 0
	for i, v := range nums {
		if 2*sum+v == total {
			return i
		}
		sum += v
	}
	return -1
}

func main() {
	nums := []int{1, 7, 3, 6, 5, 6}
	fmt.Println(pivotIndex(nums))
}