题意:
给你一个长度为 n n n的正整数序列 a a a,它包含 2 n − 1 2^n-1 2n−1个非空子序列,求所有非空子序列的权值的中位数。
一个子序列的权值为子序列内所有元素权值之和。
题解:
我们记所有元素权值的和为 s s s。
那么如果把空序列也考虑进来的话,对于任意一个权值为 x x x的子序列,一定能找到一个权值为 s − x s-x s−x的子序列。
也就是说,所有子序列的权值在 ⌊ s 2 ⌋ \lfloor \frac{s}{2}\rfloor ⌊2s⌋左右两边是对称的。
然而我们不考虑空序列,所以中位数一定是权值小于等于 ⌊ s 2 ⌋ \lfloor \frac{s}{2}\rfloor ⌊2s⌋的子序列中权值最大的那一个子序列的权值。
那么我们记一个 f f f,其中 f [ i ] f[i] f[i]代表 i i i是否是其中一个子序列的权值。
对于一个元素 x x x,转移就是 f ∣ = f < < x f|=f<<x f∣=f<<x。
时间复杂度 O ( n s 64 ) O(\frac{ns}{64}) O(64ns),然而常数极小。
代码:
#include <bits/stdc++.h>
using namespace std;
#define MAXN 2010
bitset<2000010> f;
int a[MAXN],n,sum;
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;++i)scanf("%d",a+i),sum+=a[i];
f[0]=1;
for(int i=n;i>=1;--i)f|=f<<a[i];
for(int i=(sum+1)>>1;i<=sum;++i)
if(f[i])
{
printf("%d\n",i);
return 0;
}
}