724. 寻找数组的中心索引

package com.heu.wsq.leetcode.arr;

import java.util.Arrays;

/**
 * 724. 寻找数组的中心索引
 * @author wsq
 * @date 2021/1/18
 * 给定一个整数类型的数组 nums，请编写一个能够返回数组 “中心索引” 的方法。
 * 我们是这样定义数组 中心索引 的：数组中心索引的左侧所有元素相加的和等于右侧所有元素相加的和。
 * 如果数组不存在中心索引，那么我们应该返回 -1。如果数组有多个中心索引，那么我们应该返回最靠近左边的那一个。
 *
 * 示例 1：
 * 输入：
 * nums = [1, 7, 3, 6, 5, 6]
 * 输出：3
 * 解释：
 * 索引 3 (nums[3] = 6) 的左侧数之和 (1 + 7 + 3 = 11)，与右侧数之和 (5 + 6 = 11) 相等。
 * 同时, 3 也是第一个符合要求的中心索引。
 *
 * 链接：https://leetcode-cn.com/problems/find-pivot-index
 */
public class PivotIndex {
    public int pivotIndex(int[] nums){
        int sum = Arrays.stream(nums).sum();
        int pre = 0;
        for (int i = 0; i < nums.length; i++) {
            if (sum - nums[i] == 2 * pre){
                return i;
            }
            pre += nums[i];
        }
        return -1;
    }
}