【hdu 6966】I love sequences (FWT)

题目链接

题目大意

给定序列 $[a_1,...,a_n],[b_1,...,b_n],[c_1,...,c_n]$ ，要求 ${\sum }_{p=1}^n{\sum }_{k=1}^{+\infty }{d}_{p,k}{c}_p^k$ 的值。
其中 $d_{p,k}={\sum }_{1\le i,j\le \frac{n}{p},i\oplus j=k}{a}_i{b}_j$。
其中 $\oplus$ 运算定义如下：将 $i,j$ 表示为三进制 $i=(i_{m-1}{i}_{m-2}...i_0{)}_3,j=(j_{m-1}{j}_{m-2}...j_0{)}_3$ ，则 $k=i\oplus j=(k_{m-1}{k}_{m-2}...k_0{)}_3$ ，其中 $k_t=\gcd (i_t,j_t)$。

解题思路

式子无需进一步化简，主要是求 $d_{p,k}$， 利用FWT的思路进行卷积。

对于二进制数而言，各运算卷积除满足 $C_0+C_1=(A_0+A_1)(B_0+B_1)$ 以外，还满足：

与运算：$C_1=A_1{B}_1$
或运算：$C_0=A_0{B}_0$
异或运算：$C_0-C_1=(A_0-A_1)(B_0-B_1)$
同或运算：$C_1-C_0=(A_1-A_0)(B_1-B_0)$

而题目中三进制数的运算满足：

• $C_0=A_0{B}_0$
• $C_0+C_1+C_2=(A_0+A_1+A_2)(B_0+B_1+B_2)$
• $C_0+C_2=(A_0+A_2)(B_0+B_2)$

可以先按位进行相应变换，然后做 Hadamard 积，再按位进行逆变换即可。

代码

#include <bits/stdc++.h>
#define rep(i, l, r) for (int i = l; i <= r; ++i)
#define per(i, r, l) for (int i = r; i >= l; --i)

using namespace std;
const int N = 600005;
const int mod = 1000000007;
typedef long long ll;
void fwt(ll *a, int n, int dir) {
    for (int len = 1; len * 3 <= n; len *= 3) {
        for (int i = 0; i < n; i += len * 3) {
            for (int j = i; j < i + len; j++) {
                ll &a0 = a[j], &a1 = a[j + len], &a2 = a[j + len * 2];
                if (dir == 1) {
                    a2 = (a2 + a0) % mod;
                    a1 = (a1 + a2) % mod;
                } else {
                    a1 = (a1 - a2 + mod) % mod;
                    a2 = (a2 - a0 + mod) % mod;
                }
            }
        }
    }
}
int n;
ll a[N], b[N], c[N];
ll ta[N], tb[N];
ll d[N];

int main() {
    scanf("%d", &n);
    rep(i, 1, n) scanf("%lld", &a[i]);
    rep(i, 1, n) scanf("%lld", &b[i]);
    rep(i, 1, n) scanf("%lld", &c[i]);
    int lim = 1;
    while (lim <= n) lim *= 3;
    ll ans = 0;
    rep(p, 1, n) {
        int len = 1;
        while (len <= n / p) len *= 3;
        rep(i, 0, len - 1) {
            ta[i] = (i <= n / p) ? a[i] : 0;
            tb[i] = (i <= n / p) ? b[i] : 0;
        }
        fwt(ta, len, 1);
        fwt(tb, len, 1);
        rep(i, 0, len - 1) { d[i] = ta[i] * tb[i] % mod; }
        fwt(d, len, -1);
        // printf("\n%d:\n", p);
        // rep(i, 0, len - 1) printf("%d ", d[i]);
        ll tc = 1;
        rep(i, 0, len - 1) {
            ans = (ans + d[i] * tc) % mod;
            tc = tc * c[p] % mod;
        }
    }
    printf("%lld\n", ans);
    return 0;
}