1685 有序数组中差绝对值之和(前缀和)

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

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值