bzoj 4128: Matrix BSGS+矩阵乘法

题意

给定矩阵A,B和模数p,求最小的x满足
A^x = B (mod p)
n <= 70,p <=19997，p为质数,0<= A_{ij},B_{ij}< p
保证A有逆。

分析

首先第一反应肯定是矩阵求逆+BSGS，然后用map来判断矩阵是否相等。
这样的复杂度大概是O(sqrt(p)*n^3)，还要矩阵求逆，比较麻烦。

在网上看到了几个比较好的优化。
BSGS一般都是设$A^{xw+y}\equiv B\pmod p$
这里可以设$A^{xw-y}\equiv B\pmod p$
移项$A^{xw}\equiv B*A^y\pmod p$
这样就不用矩阵求逆了。

然后判断矩阵是否相等的话，可以直接map整个矩阵，也可以把矩阵看成是p进制数然后hash，还有一种据说更快的方法是随即一个n*1的矩阵f，若A*f=B*f我们则认为这两个矩阵是相等的，这样就可以把复杂度降一阶了。

代码

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

const int N=15;

int n,mo;

struct Matrix
{
    int x,y,a[N][N];

    bool operator < (const Matrix &w) const
    {
        for (int i=1;i<=x;i++)
            if (a[i][1]<w.a[i][1]) return 0;
            else if (a[i][1]>w.a[i][1]) return 1;
        return 0;
    }

    void clear()
    {
        memset(a,0,sizeof(a));
    }

    void unit(int n)
    {
        memset(a,0,sizeof(a));x=y=n;
        for (int i=1;i<=n;i++) a[i][i]=1;
    }

}a,b,c,d,f;
map<Matrix,int> ma;

void mul(Matrix &c,Matrix a,Matrix b)
{
    c.clear();
    c.x=a.x;c.y=b.y;
    for (int i=1;i<=a.x;i++)
        for (int j=1;j<=b.y;j++)
            for (int k=1;k<=a.y;k++)
                c.a[i][j]=(c.a[i][j]+a.a[i][k]*b.a[k][j]%mo)%mo;
}

int BSGS()
{
    int w=sqrt(mo);
    c=b;
    Matrix tmp;
    mul(tmp,c,f);
    ma[tmp]=mo;
    for (int i=1;i<=w;i++)
    {
        mul(c,c,a);mul(tmp,c,f);
        ma[tmp]=i;
    }
    c.unit(n);
    for (int i=1;i<=w;i++) mul(c,c,a);
    d=c;
    for (int i=1;i<=w;i++)
    {
        mul(tmp,c,f);
        int x=ma[tmp];
        if (x) return i*w-x%mo;
        mul(c,c,d);
    }
}

int main()
{
    scanf("%d%d",&n,&mo);
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
            scanf("%d",&a.a[i][j]);
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
            scanf("%d",&b.a[i][j]);
    b.x=b.y=a.x=a.y=f.x=n;f.y=1;
    for (int i=1;i<=n;i++) f.a[i][1]=rand()%(mo-1)+1;
    printf("%d",BSGS());
    return 0;
}