2020 ICPC Universidad Nacional de Colombia Programming Contest---D

D.Dice

题意：

给定$n$个色子，每个色子有$k$个面，每个面都有一个数字，数字模$m$为$0$的面出现的概率为$0$，剩下的面出现的的概率相等，每个面在模$m$下，只有$m$中情况，我们可以算出一个色子的$m$个概率。

思路：

设$a_i$表示$1$个色子丢出的面模$m$是$i$的概率。

设$b_i$表示$1$个色子丢出的面模$m$是$i$的概率。

则：
$$\{\begin{array}{l}{b}_0=a_0\ast a_0+a_1\ast a_{m-1}+a_2\ast a_{m-2}...a_{m-1}\ast a_1\\ b_1=a_0\ast a_1+a_1\ast a_0+a_2\ast a_{m-1}...a_{m-1}\ast a_2\\ b_2=a_0\ast a_2+a_1\ast a_1+a_2\ast a_0...a_{m-1}\ast a_3\\ ...\\ b_{m-1}=a_0\ast a_{m-1}+a_1\ast a_{m-2}+a_2\ast a_{m-3}...a_{m-1}\ast a_0\end{array}$$

可以写出矩阵的形式
$$[b_0,b_1...b_{m-1}]=[a_0,a_1...a_{m-1}]\times {\left[\begin{array}{lccc}{a}_0& a_{m-1}& \cdots & a_1\\ a_1& a_0& \cdots & a_2\\ ⋮& ⋮& \ddots & ⋮\\ a_{m-1}& a_{m-2}& \cdots & a_0\end{array}\right]}^T$$
可以看出我们要用矩阵快速幂来进行优化加速。

Code：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod = 1000000007;
const ll N = 210;

ll n, k, m;

ll qpow(ll x, ll y) {
    ll ans = 1;
    while(y) {
        if(y & 1) ans = ans * x % mod;
        x = x * x % mod;
        y >>= 1;
    }
    return ans;
}

struct matrix {
    ll s[N][N];
    matrix operator * (const matrix & a) const {
        matrix tmp;
        for(int i=0; i<m; i++) {
            for(int j=0; j<m; j++) {
                tmp.s[i][j] = 0;
                for(int k=0; k<m; k++){
                    tmp.s[i][j] = (tmp.s[i][j] + s[i][k] * a.s[k][j] % mod) % mod;
                }
            }
        }
        return tmp;
    }
} A, B, ans, I;



matrix mqpow(matrix a, ll y) {
    ans = I;
    while(y) {
        if(y & 1) ans = ans * a;
        a = a * a;
        y >>= 1;
    }
    return ans;
}


int val[N];

int main() {
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif
    cin >> n >> k >> m;
    
    ll aa = qpow(1ll*k-k/m, mod-2);
    for(int i=1; i<=k; i++) {
        if(i % m == 0) continue;
        else val[i % m] = (val[i % m] + aa) % mod;
    }
    for(int i=0; i<m; i++) I.s[i][i] = 1, B.s[0][i] = val[i];
    int u = 0;
    for(int i=0; i<m; i++) {
        for(int j=0; j<m; j++) {
            A.s[i][j] = val[(j+u) % m];
        }
        u = (u - 1 + m) % m;
    }  
    
    A = mqpow(A, (n-1));
    ans = B * A;
    cout << ans.s[0][0] << endl;
    
    return 0;
}