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;
}
posted @ 2017-12-11 11:01  foreignbill  阅读(180)  评论(-1编辑  收藏  举报