leetCode---891. 子序列宽度之和【1】特别思路

891. 子序列宽度之和  https://leetcode-cn.com/problems/sum-of-subsequence-widths/

给定一个整数数组 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 。

思路：这种考虑问题的思路值得借鉴。

首先将数组排序（ 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) ) 之和为题目所求。
此题目存在问题：如果存在两个相同的元素，那么他们的序列数量就会减少，但是题目没有考虑到。

比如，2 3 3 7 给出的答案是35，但是应该是 2 3;   2 7;    3 7;   2 3 3;  2 3 7;   3 3 7;  2 3 3 7; 答案25.

class Solution {
	public:
		typedef long long LL;
		const LL MOD=1000000007;

		int sumSubseqWidths(vector<int>& A) {

			vector<long> p(20001,0);
			int temp= 1;
			for (int i=0; i<20001; i++) {
				p[i] = temp;
				temp *= 2;
				temp %= MOD;
			}


			int n=A.size();
			if(n<=1) return 0;
			sort(A.begin(),A.end());

			LL sum=0;
			LL last;
			for(int i=0; i<n; i++) {
				//asmax=2^i-1;
				last=A[i]*(p[i]-p[n-i-1])%MOD;
				sum=(sum+last+MOD)%MOD;
				//asmin=2^(n-1-i)-1;
			}
			return sum;
		}

};