D. Magic Gems（矩阵快速幂）

题目链接

题意：你有任意个魔法宝石，每个魔法宝石可以分解为m个普通宝石，每个魔法宝石和普通宝石占1单位空间。现在你有一个长度为n的空间，问有多少种方案可以填满所有空间。

题解：首先运用dp思想，f[i]代表填满i个空间的方案数，状态转移方程就是f[i] = f[i-1] + f[i-m];最后一个魔法宝石没有分解的方案数+最后一个魔法宝石分解的方案数。时间复杂度为O(n)，这样做显然不行会超时，所以要用矩阵快速幂加速。

                                  

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod = 1e9+7;
ll n,m;
struct mat{
	ll a[110][110];
};

mat mat_mul(mat x,mat y){
	mat res;
	memset(res.a,0,sizeof res.a);
	for(int i=0;i<m;i++)
	  for(int j=0;j<m;j++)
	    for(int k=0;k<m;k++)
	    res.a[i][j] = (res.a[i][j]+x.a[i][k]*y.a[k][j])%mod;
	return res;
}


void mat_pow(ll n){
	mat c,res;
	c.a[0][0]=c.a[0][m-1]=1;
	for(int i=1;i<m;i++) c.a[i][i-1]=1;
	memset(res.a,0,sizeof res.a);
	for(int i=0;i<m;i++) res.a[i][i]=1;
	while(n){
		if(n&1) res = mat_mul(res,c);
		c = mat_mul(c,c);
		n>>=1;
	}
	ll ans =0 ;
	for(int i=0;i<m;i++)
		ans = (ans+res.a[0][i])%mod;
	cout<<ans<<endl;
}


int main()
{
    cin>>n>>m;
    if(m>n) puts("1");
	else mat_pow(n-m+1); 
}