[学习笔记] 单位根反演

最新推荐文章于 2022-11-02 10:57:53 发布

ikrvxt

最新推荐文章于 2022-11-02 10:57:53 发布

阅读量930

点赞数 1

分类专栏： # 反演(莫比乌斯/二项式/单位根文章标签：概率论线性代数单位根反演

本文链接：https://blog.csdn.net/Emm_Titan/article/details/122752726

版权

反演(莫比乌斯/二项式/单位根专栏收录该内容

6 篇文章 0 订阅

订阅专栏

单位根反演

$[k\mid n]=\frac 1k\sum_{i=0}^{k-1}\omega_k^{in}$

$k$ 次单位根是 $k$ 次幂为 $1$ 的复数解 $w_k$ 。利用单位圆和单位根的关系很容易证明。

$k\mid n$

显然 $\omega_k^{in}$ ，相当于转了 $\frac{in}k$ 圈单位圆，结果仍是 $1$ 。

$\frac 1k\sum_{i=0}^{k-1}1=1=k\mid n$
$k\nmid n$

使用等比数列求和。

$[k\mid n]=\frac 1k\sum_{i=0}^{k-1}\omega_k^{in}=\frac 1k·\frac{\omega_k^{nk}-1}{\omega_k^n-1}$

$\because\omega_k^{nk}-1=\omega_k^0-1=0\wedge \omega_k^n-1\neq 0$

$\therefore \frac 1k\sum_{i=0}^{k-1}\omega_k^{in}=0=[k\mid n]=k\nmid n$

LOJ6485. LJJ学二项式定理

$T$ 组数据，给定 $n,s,a_0,a_1,a_2,a_3$ ，求 $\sum_{i=0}^nC(n,i)·s^i·a_{i\pmod 4}$ 。结果对 $998244353$ 取模。

趁热打铁，我们很容易想到将 $i$ 按模 $4$ 的余数进行分类。对于每一类我们只想统计该统计的数。
$\sum_{i=0}^n\binom nis^i\sum_{j=0}^3a_j[i\ \text{mod}\ 4=j]=\sum_{i=0}^n\binom nis^i\sum_{j=0}^3a_j[4\mid(i-j)]$
直接单位根反演： $[4|(i-j)]=\frac{1}{4}\sum_{k=0}^3\omega_{4}^{k(i-j)}$
$\Rightarrow \sum_{i=0}^n\binom nis^i·\frac 14\sum_{j=0}^3a_j·\sum_{k=0}^3\omega_4^{ki}\omega_4^{-kj}=\frac 14\sum_{j=0}^3a_j\sum_{k=0}^3\omega_4^{-kj}\sum_{i=0}^n\binom nis^i\omega_4^{ki}$
二项式定理： $\sum_{i=0}^n\binom nis^i\omega_4^{ki}=\sum_{i=0}^n\binom ni(s\omega_4^k)^i=(s\omega_4^k+1)^n$
$\Rightarrow \frac 14\sum_{j=0}^3a_j\sum_{k=0}^3\omega_4^{-jk}(s\omega_4^k+1)^n$
模 $998244353$ 意义下是有四次单位根的，且原根为 $3$ ，有 $\omega_4^1=g^{(mod-1)/4}$ 。

直接算即可。

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define mod 998244353
int T, n, s;
int a[10], w[10] = { 1, 911660635, 998244352, 86583718 };

int qkpow( int x, int y ) {
    int ans = 1;
    while( y ) {
        if( y & 1 ) ans = ans * x % mod;
        x = x * x % mod;
        y >>= 1;
    }
    return ans;
}

signed main() {
    scanf( "%lld", &T );
    while( T -- ) {
        scanf( "%lld %lld", &n, &s );
        for( int i = 0;i < 4;i ++ ) scanf( "%lld", &a[i] );
        int ans = 0;
        for( int i = 0;i < 4;i ++ ) {
            int sum = 0;
            for( int j = 0;j < 4;j ++ )
                ( sum += w[(4 - i) * j % 4] * qkpow( s * w[j] % mod + 1, n ) ) %= mod;
            ( ans += sum * a[i] ) %= mod;
        }
        printf( "%lld\n", ans * qkpow( 4, mod - 2 ) % mod );
    }
    return 0;
}
/*
6
1 2 3 4 5 6
2 3 4 5 6 1
3 4 5 6 1 2
4 5 6 1 2 3
5 6 1 2 3 4
6 1 2 3 4 5
*/