leetcode力扣刷题打卡
题目:1685. 有序数组中差绝对值之和
描述:给你一个 非递减 有序整数数组 nums 。
请你建立并返回一个整数数组 result,它跟 nums 长度相同,且result[i] 等于 nums[i] 与数组中所有其他元素差的绝对值之和。
换句话说, result[i] 等于 sum(|nums[i]-nums[j]|) ,其中 0 <= j < nums.length 且 j != i (下标从 0 开始)。
思路:
1、前缀和问题;
2、i 之前用 i * nums[i] - 前缀和;
3、i 之后用前缀和 - nums[i] * (n - 1 - i);
左右两边的值相加即可得到答案;
原代码##
class Solution {
public:
vector<int> getSumAbsoluteDifferences(vector<int>& nums) {
int n = nums.size();
vector<int>v(n, nums[0]), ans(n);
for (int i = 1; i < n; ++i) {
v[i] = v[i - 1] + nums[i];
}
for (int i = 0; i < n; ++i) {
if (!i) {
ans[i] = v[n - 1] - nums[0] * (n);
} else {
ans[i] = i * nums[i] - v[i - 1] + v[n - 1] - v[i] - nums[i] * (n - 1 - i);
}
}
return ans;
}
};
总结:
1、前缀和的变形;
2、数学计算;