[JZOJ3341]寺庙逃脱

题目大意

$Q$组询问形如：
给定$K,A,B,C,D,N,P$,给出递推式

$$f(n) = \begin{cases} K, & n=0\\ {Af(n-1)+B\over Cf(n-1)+D},& n>0 \end{cases}$$
求 $f(N)$。

$Q\le10^4,N\le 10^{18},2\le P\le10^9,-P<A,B,C,D<P$
题目保证$P$是质数，且递推式的任意一个时刻，$Cf(n-1)+D\not=0$。

---

题目分析

分式上下都有递推项，直接做很难处理，应当想办法拆开来维护。
令

$$f(n)={A_nf(0)+B_n\over C_nf(0)+D_n}$$
可以发现 $A_i,B_i,C_i,D_i$有递推式，而且都可以使用矩阵乘法维护。
至此题目就很好地解决了，时间复杂度 $\mathrm O(Q4^3log_2n)$。

---

代码实现

#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

typedef long long LL;

const int S=4;

int k,a,b,c,d,p,q,ans;
LL n;

struct matrix
{
    int r,c;
    int num[S][S];
}T,init,one;

matrix operator*(matrix a,matrix b)
{
    matrix c;
    memset(&c,0,sizeof c);
    for (int i=0;i<a.r;i++)
        for (int j=0;j<b.c;j++)
            for (int k=0;k<a.c;k++)
                (c.num[i][j]+=1ll*a.num[i][k]*b.num[k][j]%p)%=p;
    return c.r=a.r,c.c=b.c,c;
}

matrix operator^(matrix a,LL b)
{
    matrix ret=one,c=T;
    for (;b;b>>=1,c=c*c)
        if (b&1) ret=ret*c;
    return ret;
}

int quick_power(int a,int b)
{
    int ret=1,c=a;
    for (;b;b>>=1,c=1ll*c*c%p)
        if (b&1) ret=1ll*ret*c%p;
    return ret;
}

int R(int x)
{
    return quick_power(x,p-2);
}

int main()
{
    one.r=one.c=4;
    one.num[0][0]=one.num[1][1]=one.num[2][2]=one.num[3][3]=1;
    freopen("moveout.in","r",stdin);
    freopen("moveout.out","w",stdout);
    scanf("%d",&q);
    while (q--)
    {
        scanf("%d%d%d%d%d%d%d",&k,&a,&b,&c,&d,&n,&p);
        ((a%=p)+=p)%=p,((b%=p)+=p)%=p,((c%=p)+=p)%=p,((d%=p)+=p)%=p;
        memset(&init,0,sizeof init),init.num[0][0]=1,init.num[0][1]=0,init.num[0][2]=0,init.num[0][3]=1,init.r=1,init.c=4;
        memset(&T,0,sizeof T),T.r=4,T.c=4;
        T.num[0][0]=a,T.num[2][0]=b,T.num[1][1]=a,T.num[3][1]=b;
        T.num[0][2]=c,T.num[2][2]=d,T.num[1][3]=c,T.num[3][3]=d;
        init=init*(T^n);
        a=init.num[0][0],b=init.num[0][1],c=init.num[0][2],d=init.num[0][3];
        int X=(1ll*a*k%p+b)%p,Y=R((1ll*c*k%p+d)%p);
        X=1ll*X*Y%p;
        printf("%d\n",X);
    }
    fclose(stdin);
    fclose(stdout);
    return 0;
}