美文网首页
【算法题】1685. 有序数组中差绝对值之和

【算法题】1685. 有序数组中差绝对值之和

作者: 程序员小2 | 来源:发表于2023-02-06 09:56 被阅读0次

题目:

给你一个 非递减 有序整数数组 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 <= 10^5
1 <= nums[i] <= nums[i + 1] <= 10^4

java代码:

class Solution {
    public int[] getSumAbsoluteDifferences(int[] nums) {
        int sum = 0;
        int[] prefixSum = new int[nums.length];
        //计算前缀和
        for(int i=0;i<nums.length;i++){
            sum += nums[i];
            prefixSum[i] = sum;
        }
        //计算每个数的差绝对值之和
        int[] output = new int[nums.length];
        for(int i=0;i<nums.length;i++){
            // int sumOfLeftDifferences = (i+1)*nums[i]-prefixSum[i]; 
            // int sumOfRightDifferences = prefixSum[nums.length-1]-prefixSum[i]-nums[i]*(nums.length-1-i);
            // sumOfDifferences =  sumOfLeftDifferences+sumOfRightDifferences;
            output[i] = (i+1)*nums[i]-prefixSum[i]+ prefixSum[nums.length-1]-prefixSum[i]-nums[i]*(nums.length-1-i);
        }
        return output;
    }
}

相关文章

网友评论

      本文标题:【算法题】1685. 有序数组中差绝对值之和

      本文链接:https://www.haomeiwen.com/subject/sfctkdtx.html