[BZOJ4818][Sdoi2017][容斥原理][矩陣優化DP]序列計數
考慮容斥原理
Ans=f_{滿足和為p的倍數}-f_{滿足和為p的倍數求不含質數}
可以DP,
f(i,j)
表示轉移到第i位,前i位和模P等於j的方案數
那麼顯然
f(i,j)=\sum f(i-1,k)*cnt_{(j-k+p)\mod p}
其中cnt_i
表示1~m中模P為i的個數(計算第二個個數的時候要出去質數)。直接轉移當然不行,進一步可以發現這個DP可以用矩陣乘法來優化,就用一般套路就行。
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <string>
#define N 110
#define P 20170408
#define M 20000010
using namespace std;
int n,m,p;
char vis[M];
int prime[1280000];
int cnt[N];
inline void Add(int &x,int y){
if((x+=y)>=P) x-=P;
}
struct Mat{
int a[N][N];
Mat(){ for(int i=0;i<p;i++)for(int j=0;j<p;j++)a[i][j]=0; }
int *operator [](int x){ return a[x]; }
friend Mat operator *(Mat A,Mat B){
Mat C;
for(int i=0;i<p;i++)
for(int j=0;j<p;j++)
for(int k=0;k<p;k++)
Add(C[i][j],1ll*A[i][k]*B[k][j]%P);
return C;
}
}f1,f2,g;
inline Mat Pow(Mat A,int c){
Mat ret;
for(int i=0;i<p;i++) ret[i][i]=1;
for(;c;c>>=1,A=A*A) if(c&1) ret=ret*A;
return ret;
}
int main(){
freopen("1.in","r",stdin);
freopen("1.out","w",stdout);
scanf("%d%d%d",&n,&m,&p);
for(int i=1;i<=m;i++) cnt[i%p]++;
for(int i=0;i<p;i++)
for(int j=0;j<p;j++)
g[i][j]=cnt[(i-j+p)%p];
f1[0][0]=f2[0][0]=1;
f1=f1*Pow(g,n);
vis[1]=1;
for(int i=2;i<=m;i++){
if(!vis[i]) prime[++*prime]=i;
for(int j=1;j<=*prime&&1ll*prime[j]*i<=m;j++)
if(vis[prime[j]*i]=1,i%prime[j]==0) break;
}
memset(cnt,0,sizeof(cnt));
for(int i=1;i<=m;i++)
if(vis[i]) cnt[i%p]++;
for(int i=0;i<p;i++)
for(int j=0;j<p;j++)
g[i][j]=cnt[(i-j+p)%p];
f2=f2*Pow(g,n);
printf("%d\n",(f1[0][0]-f2[0][0]+P)%P);
//printf("%d\n",clock());
return 0;
}
相關文章
- 序列(dp+矩陣加速)矩陣
- #19. 計數(容斥原理)
- bzoj 3812: 主旋律 [容斥原理 狀壓DP]
- 容斥原理——數學知識
- 容斥原理
- 巨大的數(dp+矩陣加速)矩陣
- 幸運數(dp+矩陣加速)矩陣
- 【模板】容斥原理
- POJ3744 Scout YYF I (概率DP + 矩陣優化)矩陣優化
- 容斥原理講解
- 有標號DAG計數 [容斥原理 子集反演 組合數學 fft]FFT
- 遊戲裡面的容斥原理遊戲
- bzoj 2655: calc [容斥原理 伯努利數]
- BZOJ 3329 Xorequ:數位dp + 矩陣快速冪矩陣
- 容斥原理學習筆記筆記
- HDU 4059 The Boss on Mars ( 容斥原理)
- HDU 4135 Co-prime(容斥原理+分解質因數)
- HDU4390Number Sequence(容斥原理)
- HDU4407Sum ( 容斥原理)
- 演算法之DP——01矩陣演算法矩陣
- POJ 3744 概率dp+矩陣矩陣
- 動態dp & 矩陣加速遞推矩陣
- bzoj1042: [HAOI2008]硬幣購物(Dp+容斥原理)
- CUDA 矩陣乘法終極優化指南矩陣優化
- 容斥
- HDU 5468 Puzzled Elena(DFS序+容斥原理)
- HDU2841 Visible Trees (容斥原理)
- 矩陣鍵盤原理剖析矩陣
- 反射容斥反射
- BZOJ3329: Xorequ(二進位制數位dp 矩陣快速冪)矩陣
- 07:矩陣歸零消減序列和矩陣
- HDU 1695-GCD(容斥原理+尤拉函式)GC函式
- 3.5、矩陣變數矩陣變數
- 序列 DP
- THREE 矩陣優先原則和平移旋轉矩陣矩陣
- 矩陣計算矩陣
- bzoj1853: [Scoi2010]幸運數字(容斥原理)
- matlab計算含有未知數的矩陣Matlab矩陣