有一个集合S={1,2,3,4,5,6},他的一个子集的最大值是4,最小值是1,则这样的子集的个数是2^2,因为最大为4最小为1的集合最大是{1,2,3,4},1,4固定了,剩下的就是{2,3}的子集的个数了,{2,3}有2^2个子集,所以。。。。
假设一个集合,已经从小到大排序好了,则他的子集中最小为num[i],最大为num[j] (i < j) 的集合的个数为2^(j-i-1),即2^k(k=j-i-1,0<=k<=n-2)。
答案就是res += 2^(j-i-1)*(num[j]-num[i])
想到这里我就开始俩循环暴力了,果然tle,然后去看了看别人的题解,才发现优化的地方。
sum为前缀和
假设j-i-1=1,即k=1
则sum[3] - sum[1] + sum[4] - sum[2] + sum[5] - sum[3] = sum[5]+sum[4]-sum[1]-sum[2]=sum[5]-sum[2]-sum[3];
sum[3] - sum[1] + sum[4] - sum[2] + sum[5] - sum[3] + sum[n]-sum[n-2] = sum[n]-sum[k+1]-sum[n-k-1];
这样也就是把k=1时的差的绝对值都加一块了,双重循环就可以用一重的循环来搞定了。。。
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int MAXN = 3e5+10;
const ll mod = 1e9+7;
ll two[MAXN];
ll num[MAXN];
ll preSum[MAXN];
int main()
{
cin.sync_with_stdio(false);
cin.tie(false);
two[0] = 1;
for(int i = 1; i < MAXN; ++i)
two[i] = two[i-1]*2%mod;
ll res = 0;
int n;
cin >> n;
for(int i = 1; i <= n; ++i)
cin >> num[i];
sort(num+1,num+1+n);
for(int i = 1; i <= n; ++i)
preSum[i]=preSum[i-1]+num[i];
for(int i = 0; i < n-1; ++i)
{
ll temp = preSum[n]-preSum[i+1]-preSum[n-i-1];
temp %= mod;
if(temp < 0)
temp = (temp+mod)%mod;
res = (res+two[i]*temp%mod)%mod;
}
cout <<res <<endl;
return 0;
}