Bzoj5019: [Snoi2017]遗失的答案

\(L\) 唯一分解为 \(p_1^{a_1}p_2^{a_2}...p_k^{a_k}\)
对 G 也分解为 \(p_1^{b_1}p_2^{b_2}...p_k^{b_k}\)
\(a_i , b_i\) 分别为 \(p_i\) 这个质因子幂次的上下界。
显然为了满足 \(gcd\)\(G\)\(lcm\)\(L\),对于每一个 \(p_i\) ,就至少要
有一个数触其上界,有一个数触其下界。
那么就可以拿一个长为 \(2^k\) 的二进制状态表示每个质因子是否已
触其上界,是否已触其下界。
然后前后缀对于 \(L\) 的约数且是 \(G\) 的倍数的数作一遍 \(DP\)
直接 \(FWT\) 合并成为答案
还有更加优秀的容斥做法
首先问题已经转化为给定一些集合,求出或为全集的方案数
\(f_s\) 表示或为 \(s\) 的方案数
直接求不方便
\(g_s=\sum_{i\subset s}f_i\) 表示或为 \(s\) 的子集的方案数
那么 \(g_s=2^{cnt_s}-1\)(\(cnt_s\) 表示 \(s\) 的子集个数)
那么容斥得到 \(f_s=\sum_{i\subset s}(-1)^{|s|-|i|}g_i\)
对于钦定了一定要选集合 \(x\),只需要把 \(x\) 的超集 \(-1\) 然后再次做上面的容斥
记得 \(g_s=2^{cnt_s-1}\)

# include <bits/stdc++.h>
using namespace std;
typedef long long ll;
 
namespace IO {
    const int maxn(1 << 21 | 1);
 
    char ibuf[maxn], *iS, *iT, c;
    int f;
 
    inline char Getc() {
        return iS == iT ? (iT = (iS = ibuf) + fread(ibuf, 1, maxn, stdin), (iS == iT ? EOF : *iS++)) : *iS++;
    }
 
    template <class Int> inline void In(Int &x) {
        for (f = 1, c = Getc(); c < '0' || c > '9'; c = Getc()) f = c == '-' ? -1 : 1;
        for (x = 0; c >= '0' && c <= '9'; c = Getc()) x = (x << 1) + (x << 3) + (c ^ 48);
        x *= f;
    }
}
 
using IO :: In;
 
const int maxn(805);
const int maxm(16);
const int mod(1e9 + 7);
 
int n, g, l, bl, q, p[maxm], mx[maxm], mn[maxm], tot, st, cnt;
int size[1 << maxm], ans[maxn], pw[1 << maxm], bit[1 << maxm];
 
struct Fac {
    int x, y;
 
    inline bool operator <(Fac b) const {
        return x < b.x;
    }
} a[maxn];
 
inline void Inc(int &x, int y) {
    x += y;
    if (x >= mod) x -= mod;
}
 
inline void Pre_Work(int x) {
    if (x > n || x % g) return;
    register int i, v;
    a[++cnt].x = x;
    for (i = 0; i < tot; ++i) {
        v = 0;
        while (x % p[i] == 0) x /= p[i], ++v;
        if (v == mn[i]) a[cnt].y |= 1 << i;
        if (v == mx[i]) a[cnt].y |= 1 << (i + tot);
    }
}
 
int main() {
    In(n), In(g), In(l), In(q);
    register int qq, i, j, x, y, k, tp;
    if (l % g) {
        for (qq = 1; qq <= q; ++qq) puts("0");
        return 0;
    }
    x = l;
    for (i = 2; i * i <= x; ++i)
        if (x % i == 0) {
            while (x % i == 0) x /= i;
            p[tot++] = i;
        }
    if (x > 1) p[tot++] = x;
    st = 1 << (tot << 1), x = g, y = l;
    for (i = 0; i < tot; ++i) {
        while (x % p[i] == 0) x /= p[i], ++mn[i];
        while (y % p[i] == 0) y /= p[i], ++mx[i];
    }
    for (i = 1; i * i <= l; ++i)
        if (l % i == 0) {
            Pre_Work(i);
            if (i * i != l) Pre_Work(l / i);
        }
    sort(a + 1, a + cnt + 1);
    for (i = 1; i <= cnt; ++i) ++size[a[i].y];
    for (i = 1; i < st; i <<= 1)
        for (tp = i << 1, j = 0; j < st; j += tp)
            for (k = 0; k < i; ++k) size[j + k + i] += size[j + k];
    for (pw[0] = 1, j = size[st - 1], i = 1; i <= j; ++i) pw[i] = (pw[i - 1] + pw[i - 1]) % mod;
    for (i = 1; i < st; ++i) bit[i] = bit[i >> 1] + (i & 1);
    for (i = 1; i <= cnt; ++i)
        for (j = 0; j < st; ++j)
            if ((a[i].y & j) == a[i].y)
                Inc(ans[i], (bit[(st - 1) ^ j] & 1) ? mod - pw[size[j] - 1] : pw[size[j] - 1]);
    for (qq = 1; qq <= q; ++qq) {
        In(x);
        if (l % x || x % g) puts("0");
        else {
            y = lower_bound(a + 1, a + cnt + 1, (Fac){x, 0}) - a;
            a[y].x != x ? puts("0") : printf("%d\n", ans[y]);
        }
    }
    return 0;
}
posted @ 2018-11-02 14:37  Cyhlnj  阅读(246)  评论(0编辑  收藏  举报