【Leetcode】891. Sum of Subsequence Widths

题目地址：

https://leetcode.com/problems/sum-of-subsequence-widths/description/

给定一个长$n$数组$A$，其任意子序列的振幅定义为该子序列的最大值与最小值之差。求$A$的所有子序列的振幅的总和。答案模$10^9+7$后返回。

所有子序列的振幅等价于所有子集的振幅，与元素顺序无关，所以可以先将$A$排序，那么$A[i]$作为最大值的子集个数即为$2^i$，而其作为最小值的子集个数为$2^{n-1-i}$，所以答案就是$$\sum _{i=0}^{n-1}A[i](2^i-2^{n-1-i})$$代码如下：

class Solution {
 public:
  int sumSubseqWidths(vector<int>& a) {
    const int MOD = 1e9 + 7;
    sort(a.begin(), a.end());
    int n = a.size();
    long f[n];
    f[0] = 1;
    for (int i = 1; i < n; i++) f[i] = 2 * f[i - 1] % MOD;

    long res = 0;
    for (int i = 0; i < n; i++)
      res = (res + (f[i] - f[n - 1 - i]) * a[i]) % MOD;

    return (res + MOD) % MOD;
  }
};

时间复杂度$O(n\log n)$，空间$O(n)$。