LeetCode算法题-Find Pivot Index（Java实现）

这是悦乐书的第304次更新，第323篇原创

01 看题和准备

今天介绍的是LeetCode算法题中Easy级别的第172题（顺位题号是724）。给定一个整数nums数组，编写一个返回此数组的“pivot”索引的方法。我们将pivot索引定义为索引，其中索引左边的数字之和等于索引右边的数字之和。如果不存在这样的索引，我们应该返回-1。 如果有多个数据透视索引，则应返回最左侧的数据透视索引。例如：

输入：nums = [1,7,3,6,5,6]

输出：3

说明：索引3左侧的数字之和（nums [3] = 6）等于索引3右侧的数字之和。此外，3是发生这种情况的第一个索引。


输入：nums = [1,2,3]

输出：-1

说明：没有索引满足问题陈述中的条件。


注意：

• nums的长度将在[0,10000]范围内。

• 每个元素nums[i]将是[-1000,1000]范围内的整数。

本次解题使用的开发工具是eclipse，jdk使用的版本是1.8，环境是win7 64位系统，使用Java语言编写和测试。

02 第一种解法

题目的意思是找到数组中的一个元素所在索引，使该元素左右两边剩下的元素之和相等，对此我们需要将左边元素之和和右边元素之和计算出来，直到两者相等，也就找到来了该元素所在的位置，没有旧返回-1。

先将数组的所有元素之和计算出来，然后从左往右遍历数组元素，开始计算当前元素左边元素之和，因为第一位元素左边不存在元素，所以第一次计算得到的左边元素之和为0，而右边元素之和则为总和减去左边之和再减去当前元素，判断两者是否相等即可。

public int pivotIndex(int[] nums) {
    int leftsum = 0;
    int rightsum = 0;
    int sum = 0;
    for (int i=0; i<nums.length; i++) {
        sum += nums[i];
    }
    for (int i=0; i<nums.length; i++) {
        leftsum += i == 0 ? 0 : nums[i-1];
        rightsum = sum - leftsum - nums[i];
        if (leftsum == rightsum) {
            return i;
        }
    }
    return -1;
}

03 第二种解法

我们可以对上面的写法再简化下，但是思路还是一样的。同样是将总和先计算出来，然后从左往右遍历数组，另外定义一个变量，计算当前左边元素之和，在计算左边之和前，需要先判断当前元素左右两边之和是否相等，在先进行判断时，当前左边元素之和始终是上一次计算的结果，而总和减去当前元素剩下的值，如果满足题目条件，是左边元素之和的两倍。

public int pivotIndex2(int[] nums) {
    int sum = 0;
    for (int i=0; i<nums.length; i++) {
        sum += nums[i];
    }
    int currentSum = 0;
    for (int i=0; i<nums.length; i++) {
        if (sum - nums[i] == 2*currentSum) {
            return i;
        }
        currentSum += nums[i];
    }
    return -1;
}

04 小结

算法专题目前已日更超过五个月，算法题文章172+篇，公众号对话框回复【数据结构与算法】、【算法】、【数据结构】中的任一关键词，获取系列文章合集。

以上就是全部内容，如果大家有什么好的解法思路、建议或者其他问题，可以下方留言交流，点赞、留言、转发就是对我最大的回报和支持！