CodePlus 2017 12 月赛 可做题2 矩阵乘法+exgcd

题意

分析

注意到$a_1$对$a_k$贡献的系数为斐波那契数列第k-2项，$a_2$对$a_k$贡献的系数为第k-1项，那么可以把这两项求出来，那么$a_2$就一定是某个不定方程ax+by=c的解x。
设d为gcd(a,b)，先解出该方程一个解x0,y0，有个定理就是该方程的全部解是x0+(b/d)n,y0-(a/d)n，那么就很容易求出区间解的数量了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

int p,m;
LL k,l,r,a1;
struct Matrix{int a[3][3];}a;

void mul(Matrix &c,Matrix a,Matrix b)
{
    memset(c.a,0,sizeof(c.a));
    for (int i=1;i<=2;i++)
        for (int j=1;j<=2;j++)
            for (int k=1;k<=2;k++)
                (c.a[i][j]+=(LL)a.a[i][k]*b.a[k][j]%p)%=p;
}

Matrix ksm(Matrix x,LL y)
{
    Matrix ans;ans.a[1][1]=ans.a[2][2]=1;ans.a[1][2]=ans.a[2][1]=0;
    while (y)
    {
        if (y&1) mul(ans,ans,x);
        mul(x,x,x);y>>=1;
    }
    return ans;
}

void exgcd(LL &x,LL &y,LL a,LL b)
{
    if (!b) {x=1;y=0;return;}
    exgcd(x,y,b,a%b);
    LL tmp=x;x=y;y=tmp-(a/b)*y;
}

int gcd(int x,int y)
{
    if (!y) return x;
    else return gcd(y,x%y);
}

int main()
{
    int T;scanf("%d",&T);
    while (T--)
    {
        scanf("%lld%lld%lld%lld%d%d",&a1,&l,&r,&k,&p,&m);
        a1%=p;
        a.a[1][1]=0;a.a[1][2]=a.a[2][1]=a.a[2][2]=1;
        a=ksm(a,k-3);
        int f2=(a.a[1][1]+a.a[2][1])%p,f1=(a.a[1][2]+a.a[2][2])%p;
        m=(m+p-(LL)f2*a1%p)%p;
        int d=gcd(f1,p);
        if (m%d>0) {puts("0");continue;}
        LL x,y;
        exgcd(x,y,f1,p);
        x*=m/d;x%=p/d;
        if (x<0) x+=p/d;
        LL ans=0;
        if (r>=x) ans+=(r-x)/(p/d)+1;
        if (l-1>=x) ans-=(l-1-x)/(p/d)+1;
        printf("%lld\n",ans);
    }
    return 0;
}