TZOJ 4954 矩阵游戏

描述

婷婷是个喜欢矩阵的小朋友,有一天她想用电脑生成一个巨大的n行m列的矩阵(你不用担心她如何存储)。她生成的这个矩阵满足一个神奇的性质：若用F[i][j]来表示矩阵中第i行第j列的元素，则F[i][j]满足下面的递推式:
F[1][1]=1
F[i,j]=a*F[i][j-1]+b (j!=1)
F[i,1]=c*F[i-1][m]+d (i!=1)
递推式中a,b,c,d都是给定的常数。

现在婷婷想知道F[n][m]的值是多少,请你帮助她。由于最终结果可能很大，你只需要输出F[n][m]除以1,000,000,007的余数。

输入

一行有六个整数n,m,a,b,c,d。意义如题所述。

1<=N,M<=10^1000 000,a<=a,b,c,d<=10^9

输出

包含一个整数，表示F[n][m]除以1,000,000,007的余数

样例输入

3 4 1 3 2 6

样例输出

85

提示

样例中的矩阵为：

1 4 7 10
26 29 32 35
76 79 82 85

这道题主要运用到的有三个知识点：费马小定理（注意特判a==1，c==1的情况），快速幂，矩阵乘法

大体思路：由于题目输入的数据非常大，所以需要用到费马定理来优化，然后再用矩阵快速幂算出最终答案。

f[i][m]=f[i][1]*nerow^(m-1);

f[i+1][1]=f[i][m]*neline^1;

最终答案为ans*(nerow^(m-1)*neline^1)^(n-1)*nerow^(m-1)

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1e6 + 100, mod = 1e9 + 7;
char nn[N], mm[N];
LL n, m, a, b, c, d;
void cal(LL x[2][2], LL y[2][2])
{
    LL t[2][2] = {0, 0, 0, 0};
    for (int i = 0; i < 2; i++)
    {
        for (int j = 0; j < 2; j++)
        {
            for (int k = 0; k < 2; k++)
            {
                t[i][j] += x[i][k] * y[k][j];
                t[i][j] %= mod;
            }
        }
    }
    for (int i = 0; i < 2; i++)
    {
        for (int j = 0; j < 2; j++)
        {
            x[i][j] = t[i][j];
        }
    }
}
LL getnum(char s[], int fmod)
{
    LL res = 0;
    for (int i = 0; s[i]; i++)
    {
        res *= 10;
        res += s[i] - '0';

        res %= fmod;
    }
    return res;
}
int main()
{
    scanf("%s%s", nn, mm);
    cin >> a >> b >> c >> d;
    LL ans[2][2] = {1, 1, 0, 0}, tt[2][2] = {1, 0, 0, 1}, ta[2][2];//{1 0}
    LL nerow[2][2] = {a, 0, b, 1}, neline[2][2] = {c, 0, d, 1};    //{0 1}矩阵的零次幂
    if (a == 1)
        n = getnum(nn, mod);
    else
        n = getnum(nn, mod - 1);
    if (c == 1)
        m = getnum(mm, mod);
    else
        m = getnum(mm, mod - 1);
    //费马定理，特判a==1,c==1的情况
    n--;
    m--;
    while (m)
    {
        if (m & 1)
            cal(tt, nerow);
        cal(nerow, nerow);
        m >>= 1;
    }//快速幂
    for (int i = 0; i < 2; i++)
    {
        for (int j = 0; j < 2; j++)
        {
            ta[i][j] = tt[i][j];
        }
    }
    cal(tt, neline);
    while (n)
    {
        if (n & 1)
            cal(ans, tt);
        cal(tt, tt);
        n >>= 1;
    }
    cal(ans, ta);
    cout << ans[0][0] << endl;
    return 0;
}