[AGC020-C] Median Sum

洛谷传送门

Atcoder传送门

题目大意

给你N个正整数$a_1,a_2,...,a_N$。 考虑其非空子序列共有$2^N-1$个， 其元素的和分别为$s_1,s_2,s_3...s_{2^N-1}$，求$s_i$的中位数。

输入输出格式

输入格式

第一行为一个正整数$N$。

接下来一行$N$个正整数， 表示$a_i$。

输出格式

一行一个正整数， 表示$s_i$的中位数。

数据范围

• $1≤N≤2000$
• $1≤A_i≤2000$

解题分析

一道简单的思维题。

如果算上空集的情况， 那么对于每个集合$S_i$，都会有对应的包含另一半元素的$S'_i$。 所以这个情况的中位数为$\frac { \sum a_i}{2}$。

那么去掉空集后答案显然会变大。 我们做一遍背包从$\frac{\sum a_i}{2}$向上取第一个即可。

$O(\frac {N^3}{32})$贼快…

代码如下：

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <cctype>
#include <cmath>
#include <bitset>
#define R register
#define IN inline
#define gc getchar()
#define W while
#define MX 2018
template <class T>
IN void in(T &x)
{
    x = 0; R char c = gc;
    W (!isdigit(c)) c = gc;
    W (isdigit(c))
    x = (x << 1) + (x << 3) + c - 48, c = gc;
}
std::bitset <MX * MX> bt;//bitset大法好
int num, sum;
int main(void)
{
    bt[0] = true;
    int a;
    in(num);
    for (R int i = 1; i <= num; ++i)
    {
        in(a);
        bt |= bt << a, sum += a;
    }
    int mid = sum + 1 >> 1;
    for (R int i = mid; i <= sum; ++i)
    {
        if(bt[i]) printf("%d", i), exit(0);
    }
}