这是悦乐书的第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+篇,公众号对话框回复【数据结构与算法】、【算法】、【数据结构】中的任一关键词,获取系列文章合集。
以上就是全部内容,如果大家有什么好的解法思路、建议或者其他问题,可以下方留言交流,点赞、留言、转发就是对我最大的回报和支持!