[LeetCode][Kotlin] 1685. Sum of Absolute Differences in a Sorted Array

1685. Sum of Absolute Differences in a Sorted Array

You are given an integer array nums sorted in non-decreasing order.

 

Build and return an integer array result with the same length as nums such that result[i] is equal to the summation of absolute differences between nums[i] and all the other elements in the array.

 

In other words, result[i] is equal to sum(|nums[i]-nums[j]|) where 0 <= j < nums.length and j != i (0-indexed).

비내림차순으로 정렬된 정수 배열 nums가 제공됩니다. 

result[i]가 nums[i]와 배열의 다른 모든 요소 사이의 절대 차이의 합과 같도록 nums와 길이가 동일한 정수 배열 결과를 만들고 반환합니다. 

즉, result[i]는 sum(|nums[i]-nums[j]|)와 같습니다. 여기서 0 <= j < nums.length 및 j != i(0-인덱스)입니다.

 

Example 1:

Input: nums = [2,3,5]

Output: [4,3,5]

Explanation: Assuming the arrays are 0-indexed, then

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.

 

Example 2:

Input: nums = [1,4,6,8,10]

Output: [24,15,13,15,21]

 

Constraints:

• 2 <= nums.length <= 10^5
• 1 <= nums[i] <= nums[i + 1] <= 10^4

---

코드

class Solution {
    fun getSumAbsoluteDifferences(nums: IntArray): IntArray {
        val N = nums.size
        val total_sum = nums.sum()
        val answer = IntArray(nums.size)

        var left = 0
        nums.forEachIndexed { i, num ->
            val right = total_sum - left - num
            answer[i] = num * i - left + right - num * (N - 1 - i)
            left += num
        }

        return answer
    }
}