1. 问题描述:
给你一个非递减有序整数数组 nums 。请你建立并返回一个整数数组 result,它跟 nums 长度相同,且result[i] 等于 nums[i] 与数组中所有其他元素差的绝对值之和。换句话说, result[i] 等于 sum(|nums[i]-nums[j]|) ,其中 0 <= j < nums.length 且 j != i (下标从 0 开始)。
示例 1:
输入:nums = [2,3,5]
输出:[4,3,5]
解释:假设数组下标从 0 开始,那么
result[0] = |2-2| + |2-3| + |2-5| = 0 + 1 + 3 = 4,
result[1] = |3-2| + |3-3| + |3-5| = 1 + 0 + 2 = 3,
result[2] = |5-2| + |5-3| + |5-5| = 3 + 2 + 0 = 5。
示例 2:
输入:nums = [1,4,6,8,10]
输出:[24,15,13,15,21]
提示:
2 <= nums.length <= 105
1 <= nums[i] <= nums[i + 1] <= 104
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sum-of-absolute-differences-in-a-sorted-array
2. 思路分析:
① 因为题目中规定了数组是非递减有序的,所以对于第i个位置(下标从0开始)后面的位置减去当前位置的i对应的nums[i]值都是大于0的,所以我们可以计算出i后面的全部数字的和然后减去nums[i] * (len(nums) - i - 1),这个结果表示的就是nums[i]与后面位置的绝对值之和,对于前面的位置对应的nums[j]因为都是小于nums[i]的,而我们需要计算的结果是大于0的所以我们需要计算出nums[i] * i减去i位置之前的所有数字之和的结果(这个结果就是大于0的),i前后位置的相加的结果就是最终的结果了,所以可以知道本质上考查的就是前缀和的知识:当前位置i前后位置的元素之和,我们可以使用一个列表presum来记录到当前位置的前缀和,这样就可以利用这个列表计算某一个范围的区间和了,然后在循环中计算出当前的位置i对应的结果即可
② 除了①中的写法之外,我们还可以这样想,假如没有绝对值那么结果为sum - nums[i] * n(n为nums数组的长度),而i位置之后的nums[j]减去nums[i]的结果是正的,i位置之前的结果是负的,所以我们需要处理的就是i位置之前减去nums[i]负数的结果,因为需要得到正数,所以我们需要加上负数绝对值的两倍那么就为正确的答案了:total - nums[i] * n + 2 * (i * nums[i] - presum[i] + nums[i]),total为nums数组元素之和,n表示数组的长度,presum表示前缀和数组,i位置之前的之前的num[i] - nums[j](j < i)结果是大于0的相当于是负数的绝对值所以加上这个结果乘以2那么就表示的是i位置之前的数字与当前的nums[i]之差的绝对值之和
3. 代码如下:
from typing import List
class Solution:
def getSumAbsoluteDifferences(self, nums: List[int]) -> List[int]:
n = len(nums)
presum= [0] * n
presum[0] = nums[0]
for i in range(1, n):
presum[i] = presum[i - 1] + nums[i]
total = presum[n - 1]
res = list()
for i in range(n):
res.append((i + 1) * nums[i] - presum[i] + (total - presum[i]) - nums[i] * (n - i - 1))
return res
from typing import List
class Solution:
def getSumAbsoluteDifferences(self, nums: List[int]) -> List[int]:
n = len(nums)
presum = [0] * n
presum[0] = nums[0]
for i in range(1, n):
# 计算前缀和
presum[i] = presum[i - 1] + nums[i]
res, total = list(), presum[n - 1]
for i in range(n):
# 加上负数的绝对值的两倍即可
res.append(total - nums[i] * n + 2 * (i * nums[i] - presum[i] + nums[i]))
return res