首页 > 代码库 > BZOJ 4870 [Shoi2017]组合数问题 ——动态规划 矩阵乘法
BZOJ 4870 [Shoi2017]组合数问题 ——动态规划 矩阵乘法
注意到$r<k$
别问我为什么要强调。
考场上前30分水水。
然后写阶乘的时候大力$n\log {n}$预处理
本机跑的挺快的,然后稳稳的T掉了。
然后就是简单的矩阵乘法了。
#include <map> #include <cmath> #include <queue> #include <cstdio> #include <cstring> #include <iostream> #include <algorithm> using namespace std; #define F(i,j,k) for (int i=j;i<=k;++i) #define D(i,j,k) for (int i=j;i>=k;--i) #define ll long long #define mp make_pair int n,p,k,r; ll c; struct Matrix{ int x[51][51]; void init(){memset(x,0,sizeof x);return;} void buildt() {init();F(i,0,k-1)x[i][i]++,x[i][(i+1)%k]++;return;} void builds() {init();x[0][0]=1;return;} Matrix operator * (Matrix a) { Matrix ret; ret.init(); F(i,0,k-1) F(j,0,k-1) F(l,0,k-1) ret.x[i][j]=(ret.x[i][j]+(ll)x[i][l]*a.x[l][j])%p; return ret; } }A,B; int main() { scanf("%d%d%d%d",&n,&p,&k,&r); c=(ll)n*k; A.builds(); B.buildt(); for (;c;c>>=1,B=B*B) if (c&1LL) A=A*B; printf("%d\n",A.x[0][r]); }
BZOJ 4870 [Shoi2017]组合数问题 ——动态规划 矩阵乘法
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。