UVALive - 6886 Golf Bot (bitset)

题目链接：

$\quad$ UVA-6886

题目大意：

$\quad$ 给 $n$ 个数 $A[1, \ldots, n]$，$m$ 个数 $B[1, \ldots, m]$；问在 $A$ 中选 $1$ 个或 $2$ 个数，能构成 (指相加) 多少个 $B$ 中的数？

数据范围：

$\quad 1 \leq n, m \leq 2e5 \qquad 1 \leq {A_i}, {B_i} \leq 2e5$

解题思路：

$\quad$ DP的复杂度不够，只想到一个比较好的暴力，送上一发TLE；后经 cp 大佬强力优化，终于在T,W之后看到了AC！
$\quad$ 赛后发现大佬用 bitset 做的，嗯，，，bitset，了解一下？

定义：

$\quad$ 主要有三种构造方式：

bitset <16> A;
bitset <16> B (0xfa2);    //十六进制构造
bitset <16> C (string("011001"));    //字符串构造

//A: 0000000000000000
//B: 0000111110100010
//C: 0000000000011001

基本运算：

$\quad$ 什么与(&)，或(|)，非(~)，异或(^)，左移(<<)，右移(>>) 都是有的！等于(==)，不等于(!=)也都是有的。
$\quad$ C++ bitset 内部实现是非常 nice 的！很快！

基本函数：

$\quad$ bt.count() //1 的个数
$\quad$ bt.any() //返回是否有1
$\quad$ bt.none() //返回是否没有1
$\quad$ bt.set() //全部置为1
$\quad$ bt.reset() //全部置为0
$\quad$ bt.set(p) //把 p 位置为1
$\quad$ bt.reset(p) //把 p 位置为 0
$\quad$ bt.set(p, x) //把 p 位置为 x (x为0 或 1)
$\quad$ bt.flip() //全部取反
$\quad$ bt.flip(p) //把 p 位取反
$\quad$ bt.to_string() //转为 string


$\quad$ 回到题目上来；只要把选 $1$ 个数 和 选 $2$ 个数能构成的状态处理出来，跟 $B$ 数组的状态与 (&) 一下，统计 $1$ 的个数就可以了。

AC代码：

/**********************************************
 *Author*        :XzzF
 *Created Time*  : 2018/5/24 16:14:29
 *Ended  Time*  : 2018/5/24 16:23:52
*********************************************/

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

int n, m;
int a[MaxN + 5];
bitset <MaxN + 1> A, B, ans;

int main()
{
    while(scanf("%d", &n) != EOF)
    {
        for(int i = 1; i <= n; i++) {
            scanf("%d", &a[i]);
            A.set(a[i]);
        }
        scanf("%d", &m);
        for(int i = 1; i <= m; i++) {
            int x;
            scanf("%d", &x);
            B.set(x);
        }
        ans |= (B & A);    //只选1个数
        for(int i = 1; i <= n; i++)    //选2个数
            ans |= (B & (A << a[i]));
        printf("%d\n", ans.count());
        A.reset(); B.reset(); ans.reset();
    }
    return 0;
}

$\quad$ 强行打一波广告：压位卡常题必备神器，bitset，值得拥有！