求中心索引问题

求中心索引问题

给定一个整数类型的数组 nums，请编写一个能够返回数组“中心索引”的方法。

我们是这样定义数组中心索引的：数组中心索引的左侧所有元素相加的和等于右侧所有元素相加的和。

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

示例 1:

输入:
nums = [1, 7, 3, 6, 5, 6]
输出: 3
解释:
索引3 (nums[3] = 6) 的左侧数之和(1 + 7 + 3 = 11)，与右侧数之和(5 + 6 = 11)相等。
同时, 3 也是第一个符合要求的中心索引。
先说思路，看到题时会思考怎么去实现，在尽量减少多余循环的情况下，把数据的遍历控制在1/2，这样会减少时间复杂度O
定义两个指针从数据两端开始遍历相加直到两边的指针相遇结束，如果左右相加的合相等返回左指针加1，或者右边指针减1。话不多说上代码

package com.thread;

/**
 * @author 打小就很胖
 * @describ:求中心索引
 * @ClassName: CenterIndex
 */
public class CenterIndex {
    public static void main(String[] args) {
        int[] numbers = {1, 7, 3, 6, 5, 6};
        int[] numbers1 = {1,2,1,2};
        int centerIndex = getCenterIndex(numbers);
        System.out.println("一个中心索引："+centerIndex);
        int centerIndex2 = getCenterIndex(numbers1);
        System.out.println("多个中心索引："+centerIndex2);
    }

    public static  int getCenterIndex(int[] numbers){
        int index =0;
        int left = 0,leftSum=0;
        int right = numbers.length-1,rightSum=0;
        boolean lflag=true,rflag=true ;//标志判断本次循环是否要进行求和,如果没有移动,不需求和
        while (right-left>=1){
            if(lflag){
                leftSum += numbers[left];
            }
            if(rflag){
                rightSum += numbers[right];
            }
            if(leftSum == rightSum){
                if(right-left<=2){
//                    index= left+1;
                    index= right-1;//右边索引减1可以保证多个索引时返回左边的第一个
                    return index;
                }
            }
            //右边的合比左边的大,左边指针向右移动
            if(rightSum>leftSum){
                left++;
                rflag = false;
                lflag = true;
            }else if(rightSum<leftSum){//左边的合右边比的大,右边指针向左移动
                right--;
                lflag = false;
                rflag = true;
            }else {//两边合相等,同时向中间移动
                left++;
                right--;
            }
        }
        return -1;
    }
}

运行结果：

一个中心索引：3
多个中心索引：1

当然还有另外一种比较常见的解法，就是先求和，然后再从数据组的一端开始寻找，用总和减去当前的索引的值除以2等于开始索引到这个索引位置的上一个索引的合，那么当前的位置就是所谓的中心位置

public static int getCenterId(int[] numbers){
        int sum =0;
        for(int i=0;i<numbers.length;i++){
            sum +=numbers[i];
        }
        int lefSum = 0;
        for(int j=0;j<numbers.length;j++){
            if(lefSum==(sum-numbers[j])/2){
                return j;
            }else {
                lefSum +=numbers[j];
            }
        }
        return -1;
    }