-
Notifications
You must be signed in to change notification settings - Fork 17
/
PivotIndex.kt
51 lines (46 loc) · 1.52 KB
/
PivotIndex.kt
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
package com.daily.algothrim.leetcode
/**
* 724. 寻找数组的中心索引
*
* 给定一个整数类型的数组 nums,请编写一个能够返回数组 “中心索引” 的方法。
*
* 我们是这样定义数组 中心索引 的:数组中心索引的左侧所有元素相加的和等于右侧所有元素相加的和。
*
* 如果数组不存在中心索引,那么我们应该返回 -1。如果数组有多个中心索引,那么我们应该返回最靠近左边的那一个。
* */
class PivotIndex {
companion object {
@JvmStatic
fun main(args: Array<String>) {
println(PivotIndex().solution(intArrayOf(
1, 7, 3, 6, 5, 6
)))
println(PivotIndex().solution(intArrayOf(
1, 2, 3
)))
println(PivotIndex().solution(intArrayOf(
-1, -1, -1, -1, -1, -1
)))
}
}
// 输入:
// nums = [1, 7, 3, 6, 5, 6]
// 输出:3
// 解释:
// 索引 3 (nums[3] = 6) 的左侧数之和 (1 + 7 + 3 = 11),与右侧数之和 (5 + 6 = 11) 相等。
// 同时, 3 也是第一个符合要求的中心索引。
fun solution(nums: IntArray): Int {
var totalSum = 0
var subSum = 0
for (item in nums) {
totalSum += item
}
nums.forEachIndexed { index, item ->
if (subSum * 2 + item == totalSum) {
return index
}
subSum += item
}
return -1
}
}