P5004-专心OI - 跳房子【dp,矩阵乘法】
生活随笔
收集整理的這篇文章主要介紹了
P5004-专心OI - 跳房子【dp,矩阵乘法】
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
正題
題目鏈接:https://www.luogu.org/problemnew/show/P5004
題目大意
把NNN個無色格子排成一行,可以把某些格子染成黑色,但兩個黑色格子之間必須至少有MMM個無色格子,求方案數
解題思路
首先很明顯
fn=∑i=0n?m?1fif_n=\sum_{i=0}^{n-m-1}f_ifn?=i=0∑n?m?1?fi?
然后我們發現∑i=0n?m?1fi\sum_{i=0}^{n-m-1}f_i∑i=0n?m?1?fi?就是前綴和,那我們在矩陣乘法中維護一個前綴和
A=[∑i=0n?m?1fifn?mfn?m+1...fn?1fn]A=\begin{bmatrix}\sum_{i=0}^{n-m-1}f_i \\ f_{n-m} \\ f_{n-m+1} \\ ... \\ f_{n-1} \\ f_{n} \end{bmatrix}A=?????????∑i=0n?m?1?fi?fn?m?fn?m+1?...fn?1?fn???????????
然后看著轉移就好了。
codecodecode
#include<cstdio> #include<algorithm> #include<cstring> #define ll long long using namespace std; const ll Size=20,XJQ=1e9+7; struct matrix{ll a[Size][Size]; }f; ll n,m; matrix operator *(matrix a,matrix b) {matrix c;memset(c.a,0,sizeof(c.a));for(ll i=0;i<=m;i++)for(ll j=0;j<=m;j++)for(ll k=0;k<=m;k++)(c.a[i][j]+=a.a[i][k]*b.a[k][j]%XJQ)%=XJQ;return c; } matrix power(matrix f,ll b) {b--;matrix ans;memset(ans.a,0,sizeof(ans.a));for(int i=0;i<=m;i++)ans.a[0][i]=1;ans.a[0][m]++;while(b){if(b&1) ans=ans*f;f=f*f;b>>=1;}return ans; } int main() {scanf("%lld%lld",&n,&m);for(ll i=1;i<=m;i++)f.a[i][i-1]=1;f.a[0][m]=1;f.a[m][m]=1;f=power(f,n);printf("%lld",f.a[0][m]); }總結
以上是生活随笔為你收集整理的P5004-专心OI - 跳房子【dp,矩阵乘法】的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 如何获得一款个性化定制的无线机械键盘如何
- 下一篇: 无线机械键盘颜值与实力的碰撞无线机械键盘