题目链接:http://codeforces.com/contest/810/problem/C
题目大意:
告诉你一个含有n个互不相同的元素的集合,求它的所有非空子集中元素最大值与元素最小值之差的和。结果很大,输出它%1e9+7后的结果。
推倒:
先对A排序,然后不妨枚举子集的最小元素,看一下它对答案的贡献
考虑一个数列1 3 5 7 9,假设1是子集的最小元素
再考虑集合的最大元素,假设是3,那么5 7 9都不能选,总共1种情况,对答案贡献是1x2
假设是5,那么3随便选不选,7和9不能选,总共2种情况,对答案贡献是2x4
假设是7,那么3和5随便选不选,9不能选,总共4种情况,对答案贡献是4x6
由此可推出任意第i个数和第j个数(i
for(i=1;i<=n;i++)
{
for(j=i+1;j<=n;j++)
{
k=j-i-1;
ans+=b[k]*(a[j]-a[i])%mod;
ans%=mod;
}
}
这段代码在逻辑上可行,但时间复杂度为o(n^2),不能满足题目要求,因此要对规律进行进一步的优化。
可以将问题缩小化变为每个数进行多少次加法运算
其中最小的元素被减了2^n-2^0次即加了2^0-2^n次,最大的元素被加了2^n-2^0次,所以第i个数被加了2^i-2^(n-i-1)
*(另一种推倒:长度为1结果是ans1 = a[1]-a[0]+a[2]-a[1] ….a[n-1]-a[n-2] = (a[1]+a[2]+…+a[n-1] )- (a[0]+a[1]+….+a[n-2])
长度为2是结果是ans2 = a[2]-a[0]+a[3]-a[1]+…..+a[n-1]-a[n-1-2]
这样我们可以清楚的发现ans = ans1+ans2+ans(n-1) = sum[n-1]-sum[i]-sum[n-1-i-1],这里sum[i]代表前i个的和)*
最终代码如下:
#include <bits/stdc++.h>
using namespace std;
const int MAX = 3*1e5 + 7;
const int MOD = 1e9 + 7;
int a[MAX];
long long b[MAX];
int main()
{
int n;
cin >> n;
b[0] = 1;
for(int i = 0; i < n; ++i)
{
cin >> a[i];
b[i+1] = (b[i]*2) % MOD;
}
sort(a, a+n);
long long ans = 0;
for(int i = 0; i < n; ++i)
{
ans += (a[i]*(b[i]-b[n-i-1])) % MOD;
ans %= MOD;
}
cout << ans << endl;
return 0;
}