CF1106F Lunar New Year and a Recursive Sequence 线性递推 + k次剩余

已知\(f_i = \prod \limits_{j = 1}^k f_{i - j}^{b_j}\;mod\;998244353\),并且\(f_1, f_2, ..., f_{k - 1} = 1\),\(f_k = a\),已知\(f_n = m\),试求\(a\)

易知\(f_n = f_k^p\)

对于\(p\)满足递推式\(g[i] = \sum \limits_{j = 1}^k b[j] * g[i - j]\)

这是常系数线性递推,由于\(k \leq 100\),因此矩阵快速幂即可

那么问题就变为了\(f_k^p = m(\;mod\;998244353)\),求\(f_k\)

由于\(998244353\)的原根为\(3\),因此把\(m\)离散之后,可以写出方程

令\(f_k = 3^s(mod\;998244353)\),\(m = 3^t\),那么有\(3^{sp} = 3^t (mod\;998244353)\)

由欧拉定理\(sp = t(mod\;998244352)\),然后解一下这个同余方程

有解则输出,无解就无解

复杂度\(O(k^3 \log n + \sqrt{998244353})\)


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

#define de double
#define ll long long
#define ri register int
#define rep(io, st, ed) for(ri io = st; io <= ed; io ++)
#define drep(io, ed, st) for(ri io = ed; io >= st; io --)

#define gc getchar
inline int read() {
    int p = 0, w = 1; char c = gc();
    while(c > '9' || c < '0') { if(c == '-') w = -1; c = gc(); }
    while(c >= '0' && c <= '9') p = p * 10 + c - '0', c = gc();
    return p * w;
}
    
const int sid = 105;
const int mod = 998244353;
const int g = 3;

inline int mul(int a, int b) { return 1ll * a * b % mod; }
inline int fp(int a, int k) {
    int ret = 1;
    for( ; k; k >>= 1, a = mul(a, a))
        if(k & 1) ret = mul(ret, a);
    return ret;
}

int n, m, k, b[sid];
struct mtx {
    int f[sid][sid];
    mtx() {}
    mtx(int flag) {
        if(flag == 0) {
            for(int i = 0; i < k; i ++)
            for(int j = 0; j < k; j ++)
            f[i][j] = 0;
        } 
        if(flag == 1) {
            for(int i = 0; i < k; i ++)
            for(int j = 0; j < k; j ++)
            f[i][j] = (i == j) ? 1 : 0;
        }
    }
    int* operator [] (const int x) {
        return f[x];
    }
    friend mtx operator * (mtx a, mtx b) {
       mtx c(0);
        for(int i = 0; i < k; i ++)
        for(int j = 0; j < k; j ++)
        for(int p = 0; p < k; p ++)
        c[i][j] = (c[i][j] + 1ll * a[i][p] * b[p][j] % (mod - 1)) % (mod - 1);
        return c;
    }
} A, B;
    
inline mtx fp(mtx a, int k) {
    mtx ret(1);
    for( ; k; k >>= 1, a = a * a)
        if(k & 1) ret = ret * a;
    return ret;
}
    
map <int, int> H;
inline int BSGS(int A, int B) {
    H.clear(); H[1] = 0;
    int Ai = 1, Aj = 1, m = ceil(sqrt(mod));
    for(ri i = 1; i < m; i ++) Ai = 1ll * Ai * A % mod, H[1ll * Ai * B % mod] = i;
    Ai = 1ll * Ai * A % mod;
    for(ri i = 1; i <= m; i ++) {
        Aj = 1ll * Aj * Ai % mod;
        if(H[Aj]) return 1ll * i * m - H[Aj];
    }
}

inline int gcd(int a, int b) {
    return b ? gcd(b, a % b) : a;
}

inline void exgcd(int &x, int &y, int a, int b) {
    if(!b) { x = 1; y = 0; return; }
    exgcd(y, x, b, a % b); y -= a / b * x;
}
    
inline void Solve() {
    A[0][k - 1] = 1;
    rep (i, 0, k - 1) B[i][k - 1] = b[k - i];
    rep (i, 0, k - 2) B[i + 1][i] = 1;
    
    A = A * fp(B, n - k);

    int p = A[0][k - 1], t = BSGS(g, m);
    if(t % gcd(p, mod - 1)) printf("-1\n");
    else {
        int gd = gcd(p, mod - 1);
        int x, y, a = p, b = mod - 1;
        t /= gd; a /= gd; b /= gd;
        exgcd(x, y, a, b); 
        x = (x + mod - 1) % (mod - 1);
        x = 1ll * x * t % (mod - 1);
        printf("%d\n", fp(g, x));
    }
}
    
int main() {
    k = read();
    rep(i, 1, k) b[i] = read();
    n = read(); m = read();
    Solve();
    return 0;
}
上一篇:Codeforces 1106F Lunar New Year and a Recursive Sequence (数学、线性代数、线性递推、数论、BSGS、扩展欧几里得算法)


下一篇:@codeforces - 1106F@ Lunar New Year and a Recursive Sequence