BZOJ 4128

矩阵BSGS，同样处理即可。

有一个优化可以更快，判断矩阵相等可以$O(n)$。列矩阵的使用。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=71;
int n,m,P;
struct matrix{
    int a[N][N];
    int t;
    matrix(){}
    int* operator [] (int x){return a[x];}
    matrix operator * (matrix x){
        matrix tmp ;
        for (int i=0;i<n;i++)
            for (int j=0;j<n;j++){
                tmp[i][j]=0;
                for (int k=0;k<n;k++)
                    tmp[i][j]=(tmp[i][j]+a[i][k]*x[k][j])%P;
            }
        return tmp;
    }
    void operator *= (matrix x){*this = *this * x;}
    matrix operator ^ (ll x){
        matrix ret;
        for (int i=0;i<n;i++)ret[i][i]=1;
        matrix tmp = *this;
        for (;x;x>>=1,tmp*=tmp){if(x&1)ret *=tmp;}
        return ret;
    }
    void print(){
        for (int i=0;i<n;i++){
            for (int j=0;j<n;j++)
                printf("%d ",a[i][j]);
            puts("");
        }
    }
    bool operator == (const matrix x) const{
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                if(a[i][j]!=x.a[i][j]) return false;
        return true;
    }
};
struct Hash{
    #define Maxn 10007
    vector<matrix> mp[Maxn];
    void init(){
        for(int i=0;i<Maxn;i++)
            mp[i].clear();
    }
    int getkey(matrix t){
        ll ret=0;
        for(int i=0;i<n;i++)
            for(int j=0;j<n;j++)
                ret=(ret*817+t[i][j])%Maxn;
        return ret;
    }
    void insert(matrix tmp,ll i){
        int key=getkey(tmp);
        tmp.t=i;
        mp[key].push_back(tmp);
    }
    int Search(matrix t){
        int key=getkey(t);
        int mx=-1;
        // for(auto E:mp[key]){
        //  if(E==t){
        //      mx=max(mx,E.t);
        //  }
        // } 
        for(int i=0;i<(int)mp[key].size();i++){
            if(mp[key][i]==t){
                mx=max(mx,mp[key][i].t);
            }
        }
        return mx;
    }
}T;

inline ll BSGS(matrix a,matrix b,int p){//a^x%p=b
    ll l=1,r=p;
    ll m;
    while(l<=r){
        ll mid=(l+r)>>1;
        if(mid*mid>=p)
            r=mid-1,m=mid;
        else l=mid+1;
    }
    matrix tmp=b;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            tmp[i][j]%=p;
    T.init();
    for(int i=0;i<=m;i++){
        T.insert(tmp,i);
        tmp=tmp*a;
    }
    tmp=a^m;
    matrix ret;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            if(i==j) ret[i][j]=1;
            else ret[i][j]=0;
    for(int i=1;i<=m;i++){
        ret=ret*tmp;
        int id=T.Search(ret);
        if(id!=-1){
            return m*i-id;
        }
    }
    return -1;
}

int main(){
    matrix a,b;
    scanf("%d%d",&n,&P);
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            scanf("%d",&a[i][j]);
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            scanf("%d",&b[i][j]);
    cout<<BSGS(a,b,P);
    return 0;
}

转载于:https://www.cnblogs.com/foreignbill/p/8022033.html