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

题意:
给你一个 非递减 有序整数数组 nums 。

请你建立并返回一个整数数组 result,
它跟 nums 长度相同,且result[i] 等于 nums[i] 与数组中所有其他元素差的绝对值之和。

换句话说, result[i] 等于 sum(|nums[i]-nums[j]|) ,
其中 0 <= j < nums.length 且 j != i (下标从 0 开始)。

数据范围:
2 <= nums.length <= 1e5
1 <= nums[i] <= nums[i + 1] <= 1e4
解法:
令l[]为前缀和,r[]为后缀和
设下标为[0,n-1],
由于数组是非递减的,因此:
ans[i]+=r[i]-(n-1-i+1)*a[i],
ans[i]+=(i-0+1)*a[i]-l[i],

原理:
a[i]右边都是比a[i]大的数,设有x个数,那么这x个数的和减去x*a[i]就是a[i]与右边的数的绝对值差值,
左边同理.
code:
class Solution {
public:
    vector<int> getSumAbsoluteDifferences(vector<int>& a) {
        int n=a.size();
        vector<int>l(n),r(n);
        for(int i=0;i<n;i++){
            if(i==0)l[i]=a[i];
            else l[i]=l[i-1]+a[i];
        }
        for(int i=n-1;i>=0;i--){
            if(i==n-1)r[i]=a[i];
            else r[i]=r[i+1]+a[i];
        }
        vector<int>ans(n);
        for(int i=0;i<n;i++){
            ans[i]+=r[i]-(n-1-i+1)*a[i];
            ans[i]+=(i-0+1)*a[i]-l[i];
        }
        return ans;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值