题目:https://www.lydsy.com/JudgeOnline/problem.php?id=3328

单位根反演主要就是有

\( [k|n] = \frac{1}{k}\sum\limits_{i=0}^{k-1}w_{k}^{i*n} \)

如果 k | n ,转 n 下就会是 1 ;不然用等比数列求和公式可知是 0 。

一般是构造一个 \( f(x) = ( 1+x )^n \) 之类的,来求含有组合数的式子。比如

             \( \sum\limits_{i=0}^{n}C_{n}^{i*k} = \sum\limits_{i=0}^{n}C_{n}^{i}[ i | k ] \)

               \( = \frac{1}{k}\sum\limits_{i=0}^{n}C_{n}^{i}\sum\limits_{j=0}^{k-1}w_{k}^{j*i} \)

               \( = \frac{1}{k}\sum\limits_{j=0}^{k-1}\sum\limits_{i=0}^{n}C_{n}^{i}w_{k}^{j*i} \)

               \( = \frac{1}{k}\sum\limits_{j=0}^{k-1}(1+w_{k}^{j})^n \)

所以设 \( f(x) = ( 1+x )^n \) ,求 k 次 \( f( w_{k}^{j} ) \) 就行。

对于这道题,为了凑一个二项式的形式,把 \( F[i] \) 看作斐波那契递推矩阵 A 的 \( A^{i}[0][0] \) ,就有

\( ans = \frac{1}{k}\sum\limits_{j=0}^{k-1}\sum\limits_{i=0}^{n}C_{n}^{i}A^{i}w_{k}^{j*i} \)

有两个 i 次却没有 n-i 次,不能直接套。那个 \( w_{k}^{j*i} \) 的 \( w_{k}^{j} \) 与 i 无关,所以设 \( f(x) \) 的时候考虑把 \( w_{k}^{j} \) 作为 \( x \) 。

如果 \( f(x) = ( A+I*x )^n \) ,那么 \( f( w_{k}^{j} ) = \sum\limits_{i=0}^{n}C_{n}^{i}A^{i}w_{k}^{j*(n-i)} \)

想把 \( w_{k}^{j*(n-i)} \) 变成 \( w_{k}^{j*i} \) ,只要令 \( f(x) = x^{-n} ( A+I*x )^n \) ,然后求 \( f( w_{k}^{-j} ) \) 即可。

注意 n 是 long long 。

找原根是枚举 phi( mod ) 的质因子,然后看 \( g^{\frac{phi(mod)}{pri}} \) 。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=2e4+,K=;
ll n;int k,mod,g,wn;
void upd(int &x){x>=mod?x-=mod:;}
int pw(int x,int k)
{int ret=;while(k){if(k&)ret=(ll)ret*x%mod;x=(ll)x*x%mod;k>>=;}return ret;}
void fnd_g()
{
int pri[K],cnt=,p=mod-,d=p;////p=phi(mod)=mod-1!!
for(int i=;i*i<=d;i++)
if(d%i==)
{pri[++cnt]=i;while(d%i==)d/=i;}
if(d>)pri[++cnt]=d;
for(g=;;g++)
{
bool flag=;
for(int i=;i<=cnt;i++)if(pw(g,p/pri[i])==){flag=;break;}
if(flag)break;
}
}
struct Mtr{
int a[][];
Mtr(){memset(a,,sizeof a);}
Mtr operator* (const Mtr &b)const
{
Mtr c;
for(int i=;i<=;i++)
for(int k=;k<=;k++)
for(int j=;j<=;j++)
c.a[i][j]=(c.a[i][j]+(ll)a[i][k]*b.a[k][j])%mod;
return c;
}
}A,I;
Mtr pw(Mtr x,ll k)//ll!!!
{Mtr ret=I;while(k){if(k&)ret=ret*x;x=x*x;k>>=;}return ret;}
int main()
{
A.a[][]=A.a[][]=A.a[][]=;
I.a[][]=I.a[][]=;
int T;scanf("%d",&T);
while(T--)
{
scanf("%lld%d%d",&n,&k,&mod);fnd_g();
int ans=,wn=pw(g,(mod-)-(mod-)/k),ml=(mod--n%(mod-))%(mod-);
for(int i=,w=;i<k;i++,w=(ll)w*wn%mod)
{
Mtr t=A;t.a[][]+=w;t.a[][]+=w;upd(t.a[][]);upd(t.a[][]);
t=pw(t,n);
ans=(ans+(ll)t.a[][]*pw(w,ml))%mod;
}
ans=(ll)ans*pw(k,mod-)%mod;
printf("%d\n",ans);
}
return ;
}

bzoj 3328 PYXFIB——单位根反演的更多相关文章

  1. bzoj 3328 PYXFIB —— 单位根反演

    题目:https://www.lydsy.com/JudgeOnline/problem.php?id=3328 单位根反演,主要用到了 \( [k|n] = \frac{1}{k} \sum\lim ...

  2. BZOJ 3328: PYXFIB 单位根反演+矩阵乘法+二项式定理

    如果写过 LJJ 学二项式那道题的话这道题就不难了. #include <bits/stdc++.h> #define ll long long #define setIO(s) freo ...

  3. BZOJ 3328: PYXFIB 解题报告

    BZOJ 3328: PYXFIB 题意 给定\(n,p,k(1\le n\le 10^{18},1\le k\le 20000,1\le p\le 10^9,p \ is \ prime,k|(p- ...

  4. bzoj 3328: PYXFIB 数论

    3328: PYXFIB Time Limit: 10 Sec  Memory Limit: 256 MBSubmit: 130  Solved: 41[Submit][Status][Discuss ...

  5. BZOJ3328 PYXFIB 单位根反演

    题意:求 \[ \sum_{i=0}^n[k|i]\binom{n}{i}Fib(i) \] 斐波那契数列有简单的矩阵上的通项公式\(Fib(n)=A^n_{1,1}\).代入得 \[ =\sum_{ ...

  6. bzoj 3328 : PYXFIB

    Discription Input 第一行一个正整数,表示数据组数据 ,接下来T行每行三个正整数N,K,P Output T行,每行输出一个整数,表示结果 Sample Input 1 1 2 3 S ...

  7. 【BZOJ3328】PYXFIB(单位根反演,矩阵快速幂)

    [BZOJ3328]PYXFIB(单位根反演,矩阵快速幂) 题面 BZOJ 题解 首先要求的式子是:\(\displaystyle \sum_{i=0}^n [k|i]{n\choose i}f_i\ ...

  8. bzoj3328: PYXFIB(单位根反演+矩阵快速幂)

    题面 传送门 题解 我们设\(A=\begin{bmatrix}1 & 1 \\ 1 & 0\end{bmatrix}\),那么\(A^n\)的左上角就是\(F\)的第\(n\)项 所 ...

  9. UOJ#450. 【集训队作业2018】复读机 排列组合 生成函数 单位根反演

    原文链接https://www.cnblogs.com/zhouzhendong/p/UOJ450.html 题解 首先有一个东西叫做“单位根反演”,它在 FFT 的时候用到过: $$\frac 1 ...

随机推荐

  1. bzoj1627 / P2873 [USACO07DEC]泥水坑Mud Puddles

    P2873 [USACO07DEC]泥水坑Mud Puddles bfs入门. 对于坐标为负的情况,我们可以给数组下标加上$abs(min(minx,miny))$转正(根据题意判断) #includ ...

  2. CentOS日常维护及常用脚本

    [root@-.x.x xiewenming]# curl myip.ipip.net 当前 IP:42.62.x.x 来自于:中国 北京 北京 联通/电信 www.17ce.com  cdn解析网站 ...

  3. IDEA使用Git管理项目

    今天将项目使用Git管理了,IDEA. 第一步: 第二步:

  4. UVa 12034 比赛名次(递推)

    https://vjudge.net/problem/UVA-12034 题意: A.B两人赛马,最终名次有3种可能:并列第一:A第一B第二:B第一A第二.输入n,求n人赛马时最终名次的可能性的个数除 ...

  5. Minimum Depth of Binary Tree,求树的最小深度

    算法分析:递归和非递归两种方法. public class MinimumDepthofBinaryTree { //递归,树的最小深度,就是它左右子树的最小深度的最小值+1 public int m ...

  6. requestWindowFeature()的应用

    android开发中经常会在setContentView(R.layout.XXX); 前设置requestWindowFeature(XXXX). 他的意思是需要软件全屏显示.自定义标题(使用按钮等 ...

  7. classloader的演进

    classloader从1.6到1.7整体分成了两个版本.重点区别就是并行类加载. 1.6版本 protected synchronized Class loadClass(String name, ...

  8. C++常考面试题

    1 new/delete 与 malloc/free的区别 运算符是语言自身的特性,有固定的语义,编译器知道意味着什么,由编译器解释语义,生成相应的代码. 库函数是依赖于库的,一定程度上独立于语言的. ...

  9. C++的const类成员函数

    转自:http://blog.csdn.net/lihao21/article/details/8634876 我们知道,在C++中,若一个变量声明为const类型,则试图修改该变量的值的操作都被视编 ...

  10. gcd 与 扩gcd 总结

    gcd 定理的证明: 模板: ll gcd(ll a,ll b) { ) return a; else return gcd(b,a%b); } 扩gcd证明: 模板: ll extgcd(ll a, ...