AtCoder Grand Contest 020 C - Median Sum

题目大意：

$\quad$ 给 $N$ 个正整数数，$A_1, A_2, \cdots, A_N$， 可在这 $N$ 个数中选任意个数(但不能不选)，得到一个和；总共可以得到 $2^N - 1$ 个和，按照从小到大的顺序排好后，问第 $2^{N-1}$ 个数是多少(即问中位数是哪个)？

题目链接：

$\quad$ AGC020-C-Median Sum

数据范围：

$\quad$ $1 \leq N \leq 2000 \qquad 1 \leq A_i \leq 2000$

解题思路：

$\quad$ 令 $sum = \sum\limits_{i = 1}^N A_i$ ，由题意可知：$S_{2^N-1} = sum$ . 现在，任取几个数，它们的和记为 $X_i$ ，剩下的数的和记为 $Y_i$ ; 并且令 $P = min(X_i, Y_i)\, , Q = max(X_i, Y_i)$ ；可得：$P \leq Q，P + Q = S_{2^N-1} = sum$ ，由此推出 $P \leq \frac{1}{2} sum，Q \geq \frac{1}{2} sum$ ，并且可以得到 $P \in \{S_1, S_2, \cdots, S_{2^{N-1}-1} \},$$\,\,Q \in \{S_{2^{N-1}},\cdots,S_{2^N-1} \}$ (即 $P$ 在前半段， $Q$ 在后半段) ; （这里就不太懂，题解上说“assume”；我试了几个样例，还真的是这样！）

$\quad$ 然后呢，就是要找一个最小的 $Q$ ；还没结束，重点(黑科技)还在后面。

$\quad$ 一般求能构成哪些和，设计一个bool型的DP状态，$dp[i][j]$ 表示前 $i$ 个能否加成 $j$ 这个数；通过 $dp[i - 1][j - a[i]] 和 dp[i - 1][j]$ 转移。但是，空间和时间都是不够的，近乎达到了 $O(N^3)$ 的复杂度(虽然能开成一维的)。然而，大S用 bitset 解决了这个问题。

$\quad$ bitset 的用法就不再说了。当 bitset 中第 $pos$ 位为 1 时，就代表能得到 $pos$ 这个值；假设当前 bitset 的状态为 $B$ ，对于一个 整数 $x$ ,就是用或不用；如果用，可得到状态 $(B << x)$ ；如果不用，还是原来的状态 $B$ . 那么，新的状态 $B' = B \mid (B << x)$ 。题解上说，这样复杂度就变成了 $O(\frac{N^2 max(A_i)}{64})$ ，bitset 里面肯定有东西。

$\quad$ 举个例子吧！有3个数：“1 3 2”，现在到了 ‘2 ‘这个数，前面的 ‘1’, ‘3’ 能构成的和有 {0, 1, 3, 4} ; 所以 bitset 现在的状态为 “11011” (最右边是第 0 位)；如果不用 ‘2’ 这个数，状态依旧是 “11011”；如果用 ‘2’ 这个数，状态为 “1101100” (即之前能得到的和全体 +2)；最后将这两个状态 或($\mid$) 起来，就得到新的状态 “11111111” (即0~6都能构成)。

$\quad$ 哇！理解其中精妙后，真的感觉好NB啊！

AC代码：

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <iostream>
#include <queue>
#include <bitset>
using namespace std;
typedef long long LL;
const int inf = 1 << 30;
const LL INF = 1LL << 60;
const int MaxN = 2005;

int n;
bitset <MaxN * MaxN> bs;

int main()
{
    while(scanf("%d", &n) != EOF)
    {
        int sum = 0;
        bs[0] = 1;
        for(int i = 1; i <= n; i++) {
            int x;
            scanf("%d", &x);
            sum += x;
            bs = bs | (bs << x);
        }
        int ans = (sum + 1) / 2;
        while(bs[ans] == false) ans++;
        printf("%d\n", ans);
    }
    return 0;
}