Educational Codeforces Round 60 (Rated for Div. 2) D. Magic Gems

 思路:矩阵快速幂,根据递推的特性列出矩阵,直接求解,主要注意n<m的情况

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
ll n, m;
const int N = 105;
const ll mod = 1000000007;
struct mat {
    ll x[N][N];
    mat() {
        memset(x, 0, sizeof x);
    }
};
mat operator *(const mat &a, const mat &b) {
    mat ans;
    for(int i = 0; i < m; i++)
        for(int j = 0; j < m; j++)
            for(int k = 0; k < m; k++)
                ans.x[i][j] = (ans.x[i][j] + a.x[i][k] * b.x[k][j] % mod) % mod;
    return ans;
}
mat operator ^ (mat &a, ll n) {
    mat ret;
    for(int i = 0; i < m; ++i)
        ret.x[i][i] = 1;
    while(n) {
        if(n & 1)
            ret = ret * a;
        a = a * a;
        n >>= 1;
    }
    return ret;
}
int main() {
    cin >> n >> m;
    if(n < m) {
        cout << 1 << endl;
        return 0;
    }
    mat a;
    for(int i = 0; i < m - 1; i++)
        a.x[i + 1][i] = 1;
    a.x[0][0] = a.x[0][m - 1] = 1;
    mat ans = a ^ (n - m);
    ll ak = ans.x[0][0] * 2 % mod;
    for(int i = 1; i < m; i++)
        ak = (ak + ans.x[0][i]) % mod;
    cout << ak;
    return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值