【BZOJ】3240: [Noi2013]矩阵游戏

最新推荐文章于 2024-09-20 15:35:45 发布

aocong4527

最新推荐文章于 2024-09-20 15:35:45 发布

阅读量79

点赞数

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/iwtwiioi/p/4985813.html

版权

题意

给出$n, m(1 \le n, m \le 10^{1000000})$，求$f(n, m) \ \mod \ 10^9+7$

$$ \begin{cases} f(1, 1) = 1 \\ f(i, 1) = cf(i-1, m) + d \\ f(i, j) = af(i, j-1) + b & (j \neq 1) \end{cases} $$

其中$1 \le a, b, c, d \le 10^9$

分析

对于递推式$f_i = af_{i-1} + b$
当$a=1$时通项为$f_n = f_1 + (n-1) b$
当$a \neq 1$时通项为$f_n = a^{n-1} f_1 + \frac{b(a^{n-1} - 1)}{a-1}$
那么根据上式可以求出对应的系数
\[f(i, m) = xf(i, 1) + y\]
然后又得到
\[f(i, 1) = c(xf(i-1, 1) + y)+d = cxf(i-1, 1) + cy + d\]
就可以推出$f(n, 1)$，最后再逆推回$f(n, m)$即可。

题解

快速幂部分，可以根据欧拉定理$a^{\varphi(p)} \equiv 1 \pmod{p}, (a, p)=1$可以知道$a^{10^9+6} \equiv 1 \pmod{10^9+7}$
所以我们可以在读入的时候就对$n, m$模$10^9+6$然后再快速幂。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mo=1e9+7;
int ipow(int a, int b) {
    if(a>=mo) {
        a%=mo;
    }
    int x=1;
    for(; b; b>>=1, a=(ll)a*a%mo) {
        if(b&1) {
            x=(ll)x*a%mo;
        }
    }
    return x;
}
void getint(int &n, int &nn) {
    char c=getchar();
    n=nn=0;
    for(; c<'0'||c>'9'; c=getchar());
    for(; c>='0'&&c<='9'; c=getchar()) {
        n=((ll)n*10+c-'0')%mo;
        nn=((ll)nn*10+c-'0')%(mo-1);
    }
}
int main() {
    int n, m, nn, mm, a, b, c, d, ans;
    getint(n, nn);
    getint(m, mm);
    scanf("%d%d%d%d", &a, &b, &c, &d);
    int k, j;
    if(a==1) {
        k=c;
        j=((ll)c*(m-1+mo)%mo*b%mo+d)%mo;
    }
    else {
        int p=ipow(a, mm-1+(mo-1));
        k=(ll)c*p%mo;
        j=((ll)b*c%mo*(1-p+mo)%mo*ipow(1-a+mo, mo-2)%mo+d)%mo;
    }
    if(k==1) {
        ans=((ll)n*j%mo+1)%mo;
    }
    else {
        int p=ipow(k, nn);
        ans=((ll)j*ipow(1-k+mo, mo-2)%mo*(1-p+mo)%mo+p)%mo;
    }
    ans=(ans-d+mo)%mo;
    ans=(ll)ans*ipow(c, mo-2)%mo;
    printf("%d\n", ans);
    return 0;
}

转载于:https://www.cnblogs.com/iwtwiioi/p/4985813.html