LeetCode 724.寻找数组的中心下标

一、题目描述

给你一个整数数组 nums ，请计算数组的 中心下标 。

数组 中心下标 是数组的一个下标，其左侧所有元素相加的和等于右侧所有元素相加的和。

如果中心下标位于数组最左端，那么左侧数之和视为 0 ，因为在下标的左侧不存在元素。这一点对于中心下标位于数组最右端同样适用。

如果数组有多个中心下标，应该返回 最靠近左边 的那一个。如果数组不存在中心下标，返回 -1 。

二、算法思路

可以使用两个数组来记录，一个数组 rnums，lnums。

比如rnums[i] 表示nums[0]+nums[1]+...+nums[i-1],就是[0,i-1]元素的和

类似lnums[i] 表示nums[n-1]+nums[n-2]+...+nums[i+1]，就是[i+1,n-1]元素的和

三、算法代码

class Solution {
    public int pivotIndex(int[] nums) {
        //rnums: 表示是前面和
        //lnums: 表示是后面和
        int n = nums.length;
        int[] rnums =new int[n];
        int[] lnums =new int[n];

        for (int i = 1;i<n;i++){
            rnums[i] = rnums[i-1] + nums[i-1];
        }

        for (int i = n-2;i >= 0;i --){
            lnums[i] = lnums[i+1] + nums[i+1];
        }

        for (int i = 0; i < n; i++) {
            if (rnums[i] == lnums[i]){
                return i;
            }
        }
        return -1;
    }
}

四、运行结果