【LeetCode】891. 子序列宽度之和

题目描述

给定一个整数数组 A ，考虑 A 的所有非空子序列。

对于任意序列 S ，设 S 的宽度是 S 的最大元素和最小元素的差。

返回 A 的所有子序列的宽度之和。

由于答案可能非常大，请返回答案模 10^9+7。

示例：

输入：[2,1,3]
输出：6
解释：
子序列为 [1]，[2]，[3]，[2,1]，[2,3]，[1,3]，[2,1,3] 。
相应的宽度是 0，0，0，1，1，2，2 。
这些宽度之和是 6 。

提示：

1 <= A.length <= 20000
1 <= A[i] <= 20000

 

思路

首先将数组排序（ A 中共有 n 个元素），单独分析一个元素 A[i]，求该元素在所有子集中，几次作为最大值出现，几次作为最小值出现。任一子集均存在一个最大值和一个最小值，所以所有的 A[i] * 最大值之和，减去所有的 A[i] * 最小值之和即为所求。

元素 A[i] 被作为最大值的次数：比 A[i] 小的元素有 i 个，这 i 个元素能形成的子集数目为 2^i（这其中包括一个空集），将这些集合全部加上一个 A[i]，则这些集合中 A[i] 为最大元素；

元素 A[i] 被作为最大值的次数：比 A[i] 大的元素有 n-i-1 个，这 i 个元素能形成的子集数目为 2^(n-i-1)（这其中包括一个空集），将这些集合全部加上一个 A[i]，则这些集合中 A[i] 为最小元素；

对 i 从 0~n，所有 A[i] * ( 2^i - 2^(n-i-1) ) 之和为题目所求。

注意：int 是 short int 的省略，占内存的 2 个字节，表示范围为 - 32768 ~ 32767。long 是 long int 的省略，占内存的 4 个字节，表示范围为 - 2147483648 ~ 2147483647，此处也可以用 long long int。

 

代码

class Solution {
public:
    int sumSubseqWidths(vector<int>& A) {
        sort(A.begin(), A.end());
        int n = A.size();
        vector<long> p(20001,0);
        int temp= 1;
        for (int i=0; i<20001; i++){
            p[i] = temp;
            temp *= 2;
            temp %= 1000000007;
        }
        long res = 0;
        for (int i=0; i<n; i++){
            res += (p[i] - p[n - i - 1]) * A[i];
            res %= 1000000007;
        }
        return res;
    }
};

参考链接：https://blog.csdn.net/XX_123_1_RJ/article/details/81913307