python 寻找数组的中心索引_Leetcode724查找数组Python的中心索引,LeetCode724,寻找,python...

寻找数组的中心索引

这一次打卡不知道又摸了多久的鱼,周五没做题,周日补上。

题目

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

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

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

示例 1:

输入:

nums = [1, 7, 3, 6, 5, 6]

输出: 3

解释:

索引3 (nums[3] = 6) 的左侧数之和(1 + 7 + 3 = 11),与右侧数之和(5 + 6 = 11)相等。

同时, 3 也是第一个符合要求的中心索引。

示例 2:

输入:

nums = [1, 2, 3]

输出: -1

解释:

数组中不存在满足此条件的中心索引。

说明:

nums 的长度范围为 [0, 10000]。

任何一个 nums[i] 将会是一个范围在 [-1000, 1000]的整数。

解题思路

所最终求得的nums[i]两边的总和是相同的,因为从左边进行遍历,即存在公式S(nums)- nums[i] = S(leftnums),对数组进行遍历,找到满足该等式的i值,返回i。如果没有找到,则返回-1。解题步骤如下:

对传入数组用sum()函数进行求和

遍历数组,寻找满足S(nums)- nums[i] = S(leftnums)的i值

若存在,返回i,若不存在,返回-1

代码

class Solution(object):

def pivotIndex(self, nums):

S = sum(nums)

leftsum = 0

for i in range(len(nums)):

if leftsum * 2 == S - nums[i]:

return i

leftsum += nums[i]

else:

return -1

复杂度分析

时间复杂度:O(N)O(N),其中 NN 是

nums

的长度。

空间复杂度:O(1)O(1),使用了

S

leftsum

小结

今天没有用到第二种解法,所以我决定再去刷一道题弥补一下。每日一句如下:

我要再和生活死磕几年。要么我就毁灭,要么我就注定铸就辉煌。如果有一天,你发现我在平庸面前低了头,就向我开炮。

————杰克.凯鲁亚克《在路上》

你可能感兴趣的:(python,寻找数组的中心索引)