Codeforces 1017D The Wu（预处理）

题目链接：The Wu

题意

有一个元素个数为 $m$ 的 $01$ 串集合，集合内的 $01$ 串可以重复出现，且每一个串的长度都为 $n$，对于两个 $01$ 串 $s$ 和 $t$，它们产生的价值为 $v_{s,t}=\sum_{i=1}^n(s_i==t_i?w_i:0)$，现在有 $q$ 次询问，每次询问由一个长度为 $n$ 的 $01$ 串 $s$ 和一个数 $k$ 组成，询问集合中满足条件 $v_{s,t}\leq k$ 的字符串个数。

输入

第一行为 $3$ 个整数 $n,m,q~(1\leq n\leq12,1\leq q,m\leq5\times10^5)$。第二行为 $n$ 个整数 $w_1,w_2,\cdots,w_n~(0\leq w_i\leq100)$，接下去 $m$ 行每行一个长度为 $n$ 的 $01$ 字符串，接下去 $q$ 行每行由一个长度为 $n$ 的 $01$ 字符串和 $k~(0\leq k\leq100)$ 组成。

输出

对于每次询问，输出答案。

样例

输入
2 4 5 40 20 01 01 10 11 00 20 00 40 11 20 11 40 11 60
输出
2 4 2 3 4

输入
1 2 4 100 0 1 0 0 0 100 1 0 1 100
输出
1 2 1 2

题解

先将字符串转化为对应数字（$0$ 到 $2^n-1$），对于集合中的所有数字 $t$，统计每个数字出现的次数 $cnt_t$，接着对所有可能的询问 $s$（$0$ 到 $2^n-1$）都预处理出对应的价值 $v_{s,t}$，将价值为 $v_{s,t}$ 的集合中的数字 $t$ 的个数记录在 $sum[s][v_{s,t}]$ 中，即 $sum[s][v_{s,t}]=cnt_t$，由于询问中的 $k$ 最大值为 $100$，所以 $sum$ 数组的第 $2$ 维只需要开到 $100$，大于 $100$ 的可以不用处理，最后对 $sum$ 的每一次询问 $s$ 都做一遍前缀和，就能得到询问为 $s$ 时，价值小于等于 $k$ 的所有满足条件的 $t$ 的个数。预处理和查询的时间复杂度为 $O(m+2^n\times2^nn+100\times2^n+qn)$。

过题代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <climits>
#include <cstring>
#include <string>
#include <vector>
#include <list>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <bitset>
#include <algorithm>
#include <functional>
#include <iomanip>
using namespace std;

#define LL long long
int n, m, k, q;
char str[20];
int num[20], cnt[4096];
int sum[4096 + 100][100 + 20];

int main() {
    #ifdef LOCAL
    freopen("test.txt", "r", stdin);
//    freopen("test1.out", "w", stdout);
    #endif // LOCAL
    ios::sync_with_stdio(false);

    while(scanf("%d%d%d", &n, &m, &q) != EOF) {
        memset(cnt, 0, sizeof(cnt));
        memset(sum, 0, sizeof(sum));
        for(int i = 0; i < n; ++i) {
            scanf("%d", &num[i]);
        }
        for(int i = 0; i < m; ++i) {
            scanf("%s", str);
            int tmp = 0;
            for(int j = 0; j < n; ++j) {
                tmp |= (str[j] - '0') << j;
            }
            ++cnt[tmp];
        }
        for(int i = 0; i < (1 << n); ++i) {
            for(int j = 0; j < (1 << n); ++j) {
                int tmp = 0;
                for(int k = 0; k < n; ++k) {
                    if(((i >> k) & 1) == ((j >> k) & 1)) {
                        tmp += num[k];
                    }
                }
                if(tmp > 100) {
                    continue;
                }
                sum[i][tmp] += cnt[j];
            }
        }
        for(int i = 0; i < (1 << n); ++i) {
            for(int j = 1; j <= 100; ++j) {
                sum[i][j] += sum[i][j - 1];
            }
        }
        for(int i = 0; i < q; ++i) {
            scanf("%s%d", str, &k);
            int tmp = 0;
            for(int j = 0; j < n; ++j) {
                tmp |= (str[j] - '0') << j;
            }
            printf("%d\n", sum[tmp][k]);
        }
    }

    return 0;
}