P5004-专心OI - 跳房子【dp,矩阵乘法】
生活随笔
收集整理的這篇文章主要介紹了
P5004-专心OI - 跳房子【dp,矩阵乘法】
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
正題
題目鏈接:https://www.luogu.org/problemnew/show/P5004
題目大意
把NNN個(gè)無色格子排成一行,可以把某些格子染成黑色,但兩個(gè)黑色格子之間必須至少有MMM個(gè)無色格子,求方案數(shù)
解題思路
首先很明顯
fn=∑i=0n?m?1fif_n=\sum_{i=0}^{n-m-1}f_ifn?=i=0∑n?m?1?fi?
然后我們發(fā)現(xiàn)∑i=0n?m?1fi\sum_{i=0}^{n-m-1}f_i∑i=0n?m?1?fi?就是前綴和,那我們?cè)诰仃嚦朔ㄖ芯S護(hù)一個(gè)前綴和
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???????????
然后看著轉(zhuǎn)移就好了。
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]); }總結(jié)
以上是生活随笔為你收集整理的P5004-专心OI - 跳房子【dp,矩阵乘法】的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 如何获得一款个性化定制的无线机械键盘如何
- 下一篇: P1407-[国家集训队]稳定婚姻【ta