LeetCode每日一题之寻找数组的中心下标

前言：

大家好，今天是LeetCode每日一题的第三天，给大家分享的是 寻找数组的中心下标，难度系数两颗星！废话不多数，先上题目！

1.1 题目要求

题目类型: 寻找数组的中心下标

题目要求: 给定一个整数数组nums, 请编写一个能够返回数组"中心下标"的方法

注意事项:

1. 中心下标是数组的一个下标, 其左侧所有元素相加的和等于右侧所有元素相加的和
2. 如果数组不存在中心下标, 返回-1;
3. 如果数组有多个中心下标, 应该返回最靠近左边的那一个;
4. 中心下标可能出现在数组的两端

1.2 解题方法

1.2.1 使用双指针法

1.解题思路

核心：使用双指针法处理

例如, 存在一个整数数组 nums= {6,5,2,8,1,9,3}, 首先定义两个指针变量sum和total，

nums[i] + total -> <- sum - nums[i]

​ [6,5,2,8,1,9,3]

• sum初值为整个数组元素之和, sum指针从右往左移动, 其值递减当前元素；
• total初值为0, total指针从左至右移动, 其值递增当前元素；
• 当左右两个指针移动到sum和total值相等位置时停止, 该元素位置即为中心下标

2.代码实现

• 测试代码

package com.kuang.leetcode4;

import java.util.Arrays;

/**
 * @ClassName ArrayCenterIndex
 * @Description 寻找数组的中心下标
 * @Author 狂奔の蜗牛rz
 * @Date 2021/8/14
 */
public class ArrayCenterIndex {

    public static void main(String[] args) {
        //初始化整数数组
        int[] arr = {6,5,2,8,1,9,3};
        //打印输出中心坐标
        System.out.println(pivotIndex1(arr));
        /**
         * 左侧: 6+5+2=13 右侧1+9+3=13
         * 中心元素为: 8; 所以中心下标应为 3
         */
    }

    /**
     * 寻找中轴坐标 
     * @param nums 整型数组
     * @return 返回中心坐标
     */
    public static int pivotIndex1(int[] nums) {
        /**
         * sum变量为右指针, 其初值为该整数数组的所有元素之和, 该指针在数组中从右至左移动
         * 这里可以利用JDK1.8新特性:
         * 使用Arrays数组工具类, 首先将nums数组转换成流式, 然后调用其sum方法进行求和
         */
        int sum = Arrays.stream(nums).sum();
        /**
         * total变量为左指针, 其初值为0, 该指针从左侧开始移动, 从左至右计算累加元素的累加和
         */
        int total = 0;
        //使用for循环遍历nums数组
        for (int i = 0; i < nums.length; i++) {
            //total指针从左至右累加下标i元素和
            total += nums[i];
            //判断total和sum的值是否相等
            if(total == sum) {
                //若两者值相等, 将下标i(此时下标i 即是中心下标)进行返回
                return i;
            }
            //若两个指针的值不相同, 则sum减去当前下标元素值(即向左移动)
            sum = sum - nums[i];
        }
        //若数组不存在中心下标, 则返回-1
        return -1;
    }

}

• 测试结果：

结果：测试结果与预期相同！

1.2.2 改进版双指针

1.解题思路

• 由思路1我们得知：只要满足数组左侧和右侧的sum和相等, 当前元素位置就是中心下标;

• 这就意味着 只要2倍的sum和 加上 当前元素(其下标用i来表示)值 等于 所有元素总和(这里用total来表示), 当前元素i下标即为中心下标, 即满足等式 2 * sum + nums[i] = total

2.代码实现

• 测试代码

package com.kuang.leetcode4;

import java.util.Arrays;

/**
 * @ClassName ArrayCenterIndex
 * @Description 寻找数组的中心下标
 * @Author 狂奔の蜗牛rz
 * @Date 2021/8/14
 */
public class ArrayCenterIndex {

    public static void main(String[] args) {
        //初始化整数数组
        int[] arr = {6,5,2,8,1,9,3};
        /**
         * 左侧: 6+5+2=13 右侧1+9+3=13
         * 中心元素为: 8; 所以中心下标应为 3
         */
        //使用改进版 打印输出中心坐标 预测结果 3
        System.out.println("改进版测试结果:"+pivotIndex2(arr)); //测试结果 3
    }

    /**
     * 寻找中轴坐标 双指针法改进
     * @param nums 整型数组
     * @return 返回中心坐标
     */
    public static int pivotIndex2(int[] nums) {
        /**
         * total变量为该整数数组的所有元素之和
         */
        int total = Arrays.stream(nums).sum();
        /**
         * sum变量为从左至右累加元素的和, 其初值为0
         */
        int sum = 0;
        //使用for循环遍历nums数组
        for (int i = 0; i < nums.length; i++) {
            //判断total和sum的值是否相等
            if(2 * sum + nums[i] == total ) {
                //若两者值相等, 将下标i(此时下标i即是中心下标)进行返回
                return i;
            }
            //若两者值不相等, 将数组中下标i元素进行累加
            sum += nums[i];
        }
        //若数组不存在中心下标, 则返回-1
        return -1;
    }

}

• 测试结果

结果：测试结果与预期相同！

---

好了，今天LeetCode每日一题—寻找数组的中心下标到这里就结束了，欢迎大家学习和讨论，点赞和收藏！

---

参考视频链接：https://www.bilibili.com/video/BV1Ey4y1x7J3 (国内算法宝典-LeetCode算法50讲)