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+篇,公众号对话框回复【数据结构与算法】、【算法】、【数据结构】中的任一关键词,获取系列文章合集。

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

上一篇:通过maven-assembly-plugin将Springboot项目打包成tar.gz压缩包,在Linux环境可执行脚本直接安装成系统服务


下一篇:C++ STL——stack和queue