解题报告（一）C、（牛客练习赛41 F）简单数学题（数论 + FWT）（3.5）

繁凡出品的全新系列：解题报告系列 —— 超高质量算法题单，配套我写的超高质量题解和代码，题目难度不一定按照题号排序，我会在每道题后面加上题目难度指数（$1\sim 5$），以模板题难度 $1$ 为基准。

---

这样大家在学习算法的时候就可以执行这样的流程：

%
阅读我的【学习笔记】 / 【算法全家桶】学习算法 $\Rightarrow$ 阅读我的相应算法的【解题报告】获得高质量题单 $\Rightarrow$ 根据我的一句话题解的提示尝试自己解决问题 $\Rightarrow$ 点开我的详细题解链接学习巩固（好耶）
%
题单链接：【解题报告】快速沃尔什变换FWT（ICPC / CCPC / NOIP / NOI / CF / AT / NC / P / BZOJ）超高质量题解 ！！！

C、（牛客练习赛41 F）简单数学题（数论 + FWT）（3.5）

Link

https://ac.nowcoder.com/acm/contest/373/F

Problem

定义如下函数:
  f ( x ) = { 1 x = 1 max ⁡ { t   ∣   x   m o d   t = 0   & &   ∣ μ ( t ) ∣ = 1 } x ≥ 2 \ f(x)= \begin{cases} 1 & {x=1}\\ \max\{ t\ |\ {x \bmod t = 0}\ \&\&\ |\mu(t)| = 1 \}& {x \geq 2} \end{cases}  f(x)={1max{t ∣ xmodt=0 && ∣μ(t)∣=1}​x=1x≥2​

其中$\mu (t)$为莫比乌斯函数，详见：https://en.wikipedia.org/wiki/M%C3%B6bius_function

对于任何整数$x$，若有 $x=p_1^{n_1}{p}_2^{n_2}...p_k^{n_k}$ ，其中$p_i$是$x$ 第$i$ 大的质因子，则

$g(x)=p_1^{\beta (n_1)}{p}_2^{\beta (n_2)}...p_k^{\beta (n_k)}$​

$\beta (x)=\{\begin{array}{ll}0& \text{x is even}\\ 1& \text{x is odd}\end{array}$

现在定义$F(a,b,c)=g(f(a)\ast f(b)\ast f(c))$

并给出三个长度均为$n$ 的数列

$A_1{A}_2{A}_3...A_n$​

$B_1{B}_2{B}_3...B_n$​

$C_1{C}_2{C}_3...C_n$

对于某个整数$z$，如有 $z=F(A_i,B_j,C_k)(1\le i,j,k\le n)$

则称$GF(i,j,k)$为z的一种表示

设$S$ 集合为 S = { z ∣ z = F ( A i , B j , C k ) , 1 ≤ i , j , k ≤ n } S = \{ z|z=F(A_{i}, B_{j}, C_{k}),1 \leq i,j,k \leq n \} S={z∣z=F(Ai​,Bj​,Ck​),1≤i,j,k≤n} ，$|Z|$为 $S$ 中元素的个数

求 $ans={\sum }_{i=1}^{|Z|}{z}_i{k}_i$ ，其中 $z_i$ 是集合$S$ 的第$i$ 个元素，其中 $k_i$ 为 $z_i$ 的不同的表示的个数。答案对$10^9+7$ 取模。

注意：对于一个数 $z$ ，如果$z=F(A_{i_1},B_{j_1},C_{k_1}))=F(A_{i_2},B_{j_2},C_{k_2}))$ ，只要 $i_1\ne i_{2}or{j}_1\ne j_{2}or{k}_1\ne k_2$ ，那么我们说 $z$ 的这两种表示的两种不同的表示。

$1\le n\le 10^5,A_i,B_i,C_i\le 10^{18}$，保证后三行输入的所有整数每个数的最大质因子不会超过$71$。

Sample Input

2
2 3
4 6
6 10

Sample Output

72

Hint

一句话题解： 看上去比较麻烦，但是搞清楚每个函数的作用以后一步步分析即可。关键还是在于如何引出异或操作，我们的 $f$ 函数实际上得到的是一些质数的一次幂的乘积，而题目中告诉我们了所含质因数 $\le 71$ ，所以我们可以预处理一下，那么一共就只有 20 个数了，那么我们的 $f$ 函数的实际意义就变成了从 20 个质数中选择若干个质数，那么对于每一个质数来说就只有选或者不选两种情况，所以可以用状态压缩，压成一个二进制数，这样我们就可以使用位运算了 ~ 能想到这一点就会发现这几乎就是一个 FWT 的模板题了，再往后就是正常的变换了，同样的我们只需要想办法将进行位运算的数变成下标即找到一个下标的属性的序列，将他们卷起来求方案数 $k_i$ 最后求 ans 即可。

Solution

题目比较麻烦，一步一步来分析。

首先，分析 $f(x)$， $f(x)$ 求的是一个最大的 $t$ ，满足 $t$ 是 $x$ 的约数，且 $t$ 是质数一次幂的乘积。最大，所以肯定是所有质因子的乘积。

由于题目中规定质因数不会超过 $71$ ，所以我们可以先预处理出 $2$ 到 $71$ 的质数表，一共只有 $20$ 个，我们输入的数组，先将输入的每个数组处理成 $f(a[i])$，因为我们只需要知道选了哪些质数就行了，也就意味着每个质数只有 $0$ 或者 $1$ 两种选择，我们得到的 $f(x)$ 的值就可以表示成从质数表里挑选一些数的乘积，这样我们的 $f(x)$ 就可以表示成将 $x$ 质因数分解后选择质数表里的某些质数的下标压缩成一个二进制数 $mask$ 来表示这个状态 $f(x)$，也就是状态压缩，一共只有 $2^{20}=1e6$。

例如 $x=10=2\times 5$，即选择第 $0$ 个质数 $2$ ，第 $2$ 个质数 $5$ ，所以$mask=101=5$。则 num[mask] ++ ; ，其中 num 表示一个质数集，++ 意味着这个 $x$ 属于这个质数集，可表示成这个质数集的数的个数 +1。

然后分析 $g(x)$，结合 $\beta (x)$ 和 $F(a,b,c)$ 发现实际上就是将 $f(a),f(b),f(c)$ 三个数的相同质因数的次幂（ 0 或者 1 ）的异或值作为新次幂的数的乘积。而我们的 $f(a),f(b),f(c)$ 已经是被处理成了二进制的形式的数，所以直接异或就行了。

这样问题就转变成了处理素数集 $f(x)$ 之间的关系也就是集合与集合之间的异或操作，也就是经典 快速沃尔什变换(FWT)，FWT 可以将两个集合按照下标卷起来。

因为我们要统计的是方案数 $k_i$， 而显然 $z$ 的出现次数 $k_z=nu{m}_a\times nu{m}_b\times nu{m}_c$，其中 $a\oplus b\oplus c=z$。

即：

$$k_z=\sum _{z=a\oplus b\oplus c}nu{m}_a\times nu{m}_b\times nu{m}_c$$

实际意义就是 $a_i$ 的个数是 $nu{m}_{a_i}$， $b_i$ 的个数是 $nu{m}_{b_i}$，那么 $a_i$ 和 $b_i$ 异或起来组成的 $z$ 的 方案数 即题目中要求的不同的表示的个数 $k_i$ 就是 $nu{m}_{a_i}\times nu{m}_{b_i}$。

求出 $k_i$ ，答案为 $ans={\sum }_{i=1}^{|Z|}{z}_i{k}_i$，我们枚举状压的那个二进制数，将枚举到的 $i$ 还原为 $z$ ，加上 $z\times k_i$ （没有这个 $z$ 的话 $k_i=0$ 不影响）即可。

是不是很简单

Code

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>

using namespace std;
typedef long long ll;
typedef int itn;
const int N = (1 << 20) + 7, mod = 1e9 + 7;
const ll INF = 4e18;
itn dcnt;
#define de(x) cout << x << " x" << endl
#define de1() cout << ++ dcnt << "ok" << endl
itn n, m;
ll a[N], b[N], c[N], k[N];
ll primes[N], cnt_primes;
int cnt;
bool vis[N];

void init()
{
    for(int i = 2; i <= 71; ++ i) {
        if(vis[i] == 0) primes[cnt_primes ++ ] = 1ll * i;
        for(int j = 0; j < cnt_primes && primes[j] * i <= 71; ++ j) {
            vis[i * primes[j]] = true;
            if(i % primes[j] == 0) break;
        }
    }
}

ll qpow(ll a, ll b)
{
    ll res = 1;
    while(b) {
        if(b & 1) res = 1ll * res * a % mod;
        a = 1ll * a * a % mod;
        b >>= 1;
    }
    return res % mod;
}

int inv2 = qpow(2, mod - 2);

void XOR(ll *f, int n, int x = 1)
{
    for(int o = 2; o <= n; o <<= 1) {
        for(int i = 0, k = o >> 1; i < n; i += o) {
            for(int j = 0; j < k; ++ j) {
                ll X = f[i + j];
                ll Y = f[i + j + k];
                f[i + j] = (1ll * X + Y) % mod;
                f[i + j + k] = (1ll * (X - Y) % mod + mod) % mod;
                if(x == -1) {
                    f[i + j] = (1ll * f[i + j] * inv2) % mod;
                    f[i + j + k] = (1ll * f[i + j + k] * inv2) % mod;
                }
            }
        }
    }
}

void solve(ll *a)//求 f(a)
{
    ll x;
    for(int i = 1; i <= n; ++ i) {
        scanf("%lld", &x);
        ll cnt = 0, mask = 0;
        for(int j = 0; j < cnt_primes && x > 1; ++ j) {
            cnt = 0;
            ll p = primes[j];
            while(x % p == 0) {
                x /= p;
                cnt ++ ;
            }
            if(cnt)
                mask |= 1 << j;//选择了 j 这个数
        }
        a[mask] ++ ;
    }
}

ll cal(int x)
{
    ll res = 1;
    for(int i = 0; i < cnt_primes; ++ i) {
        if(x & (1 << i)) res = 1ll * res * primes[i] % mod;
    }
    return res;
}

int main()
{
    init();
    scanf("%d", &n);
    solve(a), solve(b), solve(c);
    XOR(a, 1 << 20), XOR(b, 1 << 20), XOR(c, 1 << 20);
    for(int i = 0; i < 1 << 20; ++ i)
        k[i] = 1ll * a[i] * b[i] % mod * c[i] % mod;
    XOR(k, 1 << 20, -1);
    ll ans = 0;
    for(int i = 0; i < 1 << 20; ++ i) {
        ans = (1ll * ans + cal(i) * k[i] % mod) % mod;
    }
    printf("%lld\n", ans);
    return 0;
}