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

最新推荐文章于 2022-05-22 00:40:02 发布

原创最新推荐文章于 2022-05-22 00:40:02 发布 · 127 阅读

0 ·

CC 4.0 BY-SA版权

go 专栏收录该内容

35 篇文章

订阅专栏

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

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))
}