Time Limit: 10 Sec Memory Limit: 128 MB
Submit: 813 Solved: 442
给定矩阵A,B和模数p,求最小的x满足
A^x = B (mod p)
第一行两个整数n和p,表示矩阵的阶和模数,接下来一个n * n的矩阵A.接下来一个n * n的矩阵B
输出一个正整数,表示最小的可能的x,数据保证在p内有解
2 7 1 1 1 0 5 3 3 2
4
对于100%的数据,n <= 70,p <=19997,p为质数,0<= A_{ij},B_{ij}< p
保证A有逆
裸的BSGS,把$x$分解为$im - j$
原式化为$a^{im} \equiv ba^j \pmod p$
其中$m = \ceil{sqrt(p)}$
然后枚举一个$j$,存到map里
再枚举一个$i$判断即可
一开始map写成bool类型了调了半个小时
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<map>
//#define LL long long
using namespace std;
const int MAXN = 4 * 1e5 + 10;
inline int read() {
char c = getchar(); int x = 0, f = 1;
while(c < '0' || c > '9') {if(c == '-') f = -1; c = getchar();}
while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
return x * f;
}
int N, mod, M;
struct Matrix {
int m[71][71];
Matrix operator * (const Matrix &rhs) const {
Matrix ans = {};
for(int i = 1; i <= N; i++)
for(int j = 1; j <= N; j++)
for(int k = 1; k <= N; k++)
(ans.m[i][j] += m[i][k] * rhs.m[k][j]) %= mod;
return ans;
}
void init() {
for(int i = 1; i <= N; i++)
for(int j = 1; j <= N; j++)
m[i][j] = read();
}
void print() {
for(int i = 1; i <= N; i++, puts(""))
for(int j = 1; j <= N; j++)
printf("%d ", m[i][j]);
}
bool operator < (const Matrix &rhs) const {
for(int i = 1; i <= 70; i++)
for(int j = 1; j <= 70; j++) {
if(m[i][j] < rhs.m[i][j]) return 1;
if(m[i][j] > rhs.m[i][j]) return 0;
}
return 0;
}
}A, B;
map<Matrix, int> mp;
void MakeMap() {
Matrix a = B;
mp[a] = 0;
for(int i = 1; i <= M; i++) a = a * A, mp[a] = i;
}
void FindAns() {
Matrix a, am = A;
for(int i = 1; i <= M - 1; i++) am = am * A;
a = am;
for(int i = 1; i <= M; i++) {
if(mp[a]) printf("%d", i * M - mp[a]), exit(0);
a = a * am;
}
}
main() {
N = read(); mod = read();
A.init(); B.init();
M = (double)ceil(sqrt(mod));
MakeMap();
FindAns();
}