寻找数组的中心索引-LeetCode

题目描述：

给定一个整数类型的数组 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
解释: 
数组中不存在满足此条件的中心索引。

解题思路：

1. 写一个工具方法，用以计算给定数组指定区间元素的和。
2. 初始化一个中间索引，计算该索引左右区间的元素和是否相等，若相等，则返回当前值，若不相等，则继续向后判断。
3. 如果数组的第二个元素到最后一个元素的和为0，则返回0；如果数组第一个元素到倒数第二个元素的和为0，则返回最后一个元素的下标。

代码实现：

class Solution {
    public int pivotIndex(int[] nums) {
        int n = nums.length;
        if (n <= 2) {
            return -1;
        }
        int index = 1;
        if (Add(nums,1,n-1) == 0) {
            return 0;
        }
        while (index+1 < n) {
            if (Add(nums,0,index-1) == Add(nums,index+1,n-1)) {
                return index;
            }
            index++;
        }
        if (Add(nums,0,n-2) == 0) {
            return n-1;
        }
        return -1;
    }

    /**
     * 计算给定数组中指定区间元素的和
     * @param arr 给定数组
     * @param l 给定区间的左起点
     * @param r 给定区间的右终点
     * @return
     */
    public int Add(int[] arr, int l, int r) {
        int sum = 0;
        for (int i = l; i <= r; i++) {
            sum += arr[i];
        }
        return sum;
    }
}