LeetCode 891. Sum of Subsequence Widths

题目描述

891. 子序列宽度之和

AC代码

将数组 A 从小到大排序。
假如我们固定了当前子序列的最小值 A[i]，假设数组下标从 0 开始，则以 A[i]为最小值贡献的答案为，$(A[n-1]-A[i])2^n-i-2+(A[n-2]-A[i])2^n-i-3+\cdots +(A[i+1]-A[i])2^0$。将括号打开，一部分为 $-A[i]\cdot (2^n-i-1-1)$，另一部分是 $S(i)=\sum n-1k=i+1A[k]\cdot 2k-i-1$。
第二部分不能每次枚举计算，但可以找到递推公式，即$S(i)=2S(i+1)+A[i+1]$，其中 $S(n-1)=0$。
然后我们可以从 $i=n-2$开始到 $i=0$枚举统计答案即可。

class Solution {
    int mod=1000000007;
    public int sumSubseqWidths(int[] A) {
        Arrays.sort(A);
        long res=0,p=1,sum=0;
        for(int x:A){
            res=(res+x*(p-1)-sum)%mod;
            sum=(sum*2+x)%mod;
            p=p*2%mod;
        }
        return (int)(res+mod)%mod;
    }
}