为什么会变成这样呢? #4(bitset)

发布时间 2023-08-13 16:39:39作者: 方而静

bitset 维护01可行背包

给定 \(n\) 个数 \(a_1,\dots,a_n\),已知 \(\sum a_i=O(n)\),求一种把 \(a\) 分为两组的方案使得两组的和尽可能接近。

期望复杂度:\(O(n\sqrt n\log n/\omega)\)

解答

\(a_i\) 中相同的数合并,容易发现只有 \(O(\sqrt n)\) 种不同的 \(a_i\),然后按照多重背包动态规划即可。

使用 bitset 优化即可做到 \(O(n\sqrt n\log n/\omega)\) 的时间复杂度。

动态 bitset

\(T\) 次询问,每次给定 \(n\) 个数 \(a_1,\dots,a_n\),已知 \(\sum a_i=O(n)\),求一种把 \(a\) 分为两组的方案使得两组的和尽可能接近。

期望复杂度:\(O(T+\sum n\sqrt n\log n/\omega)\)

解答:template trick

做法和上一个问题一样,但是必须保证每次询问的 bitset 大小都在 \(O(n)\) 才能保证正确的复杂度,这似乎意味着我们需要动态大小的 bitset。但可惜的是 C++ STL 中并没有支持动态大小的 bitset。

难道只能手写了吗?并非如此。一个最基本的想法是,我们每 \(2\) 的幂的长度(也即 \(2,4,8,16,\dots\))次开一个 bitset,每次询问时选择长度合适的 bitset 使用。当然,我们不希望写太多重复的代码,而是使用模版来自动完成这个过程:

template<int len = 1>
int calc(int n) {
  if (n > len)
    return calc<std::min(len, maxn)>(n);

  // 实际计算
  bitset<len> f;
  // ...
}