2574-左右元素和的差值

Raphael Liu Lv10

给你一个下标从 0 开始的整数数组 nums ,请你找出一个下标从 0 开始的整数数组 answer ,其中:

  • answer.length == nums.length
  • answer[i] = |leftSum[i] - rightSum[i]|

其中:

  • leftSum[i] 是数组 nums 中下标 i 左侧元素之和。如果不存在对应的元素,leftSum[i] = 0
  • rightSum[i] 是数组 nums 中下标 i 右侧元素之和。如果不存在对应的元素,rightSum[i] = 0

返回数组 answer

示例 1:

**输入:** nums = [10,4,8,3]
**输出:** [15,1,11,22]
**解释:** 数组 leftSum 为 [0,10,14,22] 且数组 rightSum 为 [15,11,3,0] 。
数组 answer 为 [|0 - 15|,|10 - 11|,|14 - 3|,|22 - 0|] = [15,1,11,22] 。

示例 2:

**输入:** nums = [1]
**输出:** [0]
**解释:** 数组 leftSum 为 [0] 且数组 rightSum 为 [0] 。
数组 answer 为 [|0 - 0|] = [0] 。

提示:

  • 1 <= nums.length <= 1000
  • 1 <= nums[i] <= 105

模拟。按要求计算前缀和 leftSum 与后缀和 rightSum,然后计算答案。

代码实现时,可以把所有元素和计算出来,当作 rightSum,然后一边遍历 nums,一边更新 leftSum 和 rightSum,同时把结果直接记到 nums}[i] 中。

[sol1-Python3]
1
2
3
4
5
6
7
8
class Solution:
def leftRightDifference(self, nums: List[int]) -> List[int]:
left_sum, right_sum = 0, sum(nums)
for i, x in enumerate(nums):
right_sum -= x
nums[i] = abs(left_sum - right_sum)
left_sum += x
return nums
[sol1-Go]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
func leftRightDifference(nums []int) []int {
rightSum := 0
for _, x := range nums {
rightSum += x
}
leftSum := 0
for i, x := range nums {
rightSum -= x
nums[i] = abs(leftSum - rightSum)
leftSum += x
}
return nums
}

func abs(x int) int { if x < 0 { return -x }; return x }

复杂度分析

  • 时间复杂度:O(n),其中 n 为 nums 的长度。
  • 空间复杂度:O(1)。仅用到若干额外变量。

欢迎关注【biIibiIi@灵茶山艾府】 ,高质量算法教学,持续更新中~

 Comments
On this page
2574-左右元素和的差值