Codeforces 1106F Lunar New Year and a Recursive Sequence (數學、線性代數、線性遞推、數論、BSGS、擴展歐幾裏得算法)
阿新 • • 發佈：2019-02-05
recursive ati 滿足 name cstring rec 其中 lun 小時
哎呀大水題。。我寫了一個多小時。。好沒救啊。。
數論板子X合一？
註意: 本文中變量名稱區分大小寫。
題意: 給一個\(n\)階遞推序列\(f_k=\prod^{n}_{i=1} f_{k-i}b_i\mod P\)其中\(P=998244353\), 輸入\(b_1,b_2,...,b_n\)以及已知\(f_1,f_2,...,f_{n-1}=1\), 再給定一個數\(m\)和第\(m\)項的值\(f_m\), 求出一個合法的\(f_n\)值使得按照這個值遞推出來的序列滿足第\(m\)項的值為給定的\(f_m\).
題解: 首先一個顯然的結論是\(f_m\)可以表示成\(\prod^{n}_{i=1} f_i^{a_i}\)
求出來\(a=A(m)\)之後這題就變成了，\(f_m=f_n^a\mod P\), 已知\(f_m, a\), 求出一組合法的\(f_n\).
根據常識，\(998244353\)有原根\(3\), 我們下文令\(G=3\) (實際上任何一個原根均可). 設\(f_m=G^p, f_n=G^q\), 則有\(G^p\equiv (G^q)^a (\mod P)\), \(p\equiv qa(\mod P-1)\), 然後用BSGS求離散對數\(p\), exgcd解出\(q\)就可以了啊……
時間復雜度\(O(\sqrt P\log P+n^3\log P)\)
坑: 註意解同余方程的時候那個\(P\)的系數不要設成負的。
代碼
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<map>
#define llong long long
using namespace std;
const int N = 100;
const int G = 3;
const int P = 998244353;
llong quickpow(llong x,llong y)
{
llong cur = x,ret = 1ll;
for(int i=0; y; i++)
{
if(y&(1ll<<i)) {y-=(1ll<<i); ret = ret*cur%P;}
cur = cur*cur%P;
}
return ret;
}
struct Matrix
{
llong a[N+3][N+3]; int sz1,sz2,sz;
void init() {for(int i=1; i<=sz1; i++) for(int j=1; j<=sz2; j++) a[i][j] = 0ll;}
Matrix() {}
Matrix(int _sz) {sz = sz1 = sz2 = _sz; init();}
Matrix(int _sz1,int _sz2) {sz1 = _sz1,sz2 = _sz2; init();}
void uinit(int _sz) {sz = sz1 = sz2 = _sz; for(int i=1; i<=sz; i++) for(int j=1; j<=sz; j++) a[i][j] = (i==j)?1:0;}
void output() {for(int i=1; i<=sz1; i++) {for(int j=1; j<=sz2; j++) printf("%lld ",a[i][j]); puts("");}}
};
Matrix operator *(Matrix x,Matrix y)
{
Matrix ret = Matrix(x.sz1,y.sz2);
for(int i=1; i<=x.sz1; i++)
{
for(int j=1; j<=x.sz2; j++)
{
for(int k=1; k<=y.sz2; k++)
{
ret.a[i][j] = (ret.a[i][j]+x.a[i][k]*y.a[k][j])%(P-1ll);
}
}
}
return ret;
}
Matrix mquickpow(Matrix x,llong y)
{
Matrix cur = x,ret; ret.uinit(x.sz);
for(int i=0; y; i++)
{
if(y&(1ll<<i)) {y-=(1ll<<i); ret = ret*cur;}
cur = cur*cur;
}
return ret;
}
namespace BSGS
{
const int B = 31595;
map<llong,int> mp;
void init()
{
llong bs = quickpow(G,B); llong j = 1ll;
for(int i=0; i<=P; i+=B,j=(j*bs)%P)
{
mp[j] = i;
}
}
llong Logarithm(llong x)
{
llong j = 1ll;
for(int i=0; i<=B; i++,j=(j*G)%P)
{
llong tmp = x*j%P;
if(mp.count(tmp))
{
llong ret = (mp[tmp]-i+(P-1))%(P-1);
return ret;
}
}
return P-1;
}
}
Matrix mA,mB,mC;
llong a[N+3],b[N+3];
int n; llong m,p,q,lq,lx;
llong gcd(llong x,llong y) {return y==0 ? x : gcd(y,x%y);}
void exgcd(llong _a,llong _b,llong &_x,llong &_y)
{
if(_b==0ll) {_x = 1ll,_y = 0ll; return;}
exgcd(_b,_a%_b,_x,_y);
llong tmp = _x; _x = _y; _y = tmp-(_a/_b)*_y;
}
llong CongruenceEquation(llong _a,llong _b,llong mod)
{
llong g = gcd(_a,mod),x,y;
exgcd(_a/g,mod/g,x,y);
return (x*(_b/g)%mod+mod)%mod;
}
int main()
{
BSGS::init();
scanf("%d",&n);
for(int i=1; i<=n; i++) scanf("%I64d",&b[i]);
scanf("%I64d",&m);
mA = Matrix(1,n); mA.a[1][1] = 1ll;
mB = Matrix(n); for(int i=1; i<n; i++) mB.a[i][i+1] = 1ll; for(int i=1; i<=n; i++) mB.a[i][1] = b[i];
mC = mA*mquickpow(mB,m-n); p = mC.a[1][1]; scanf("%I64d",&q);
lq = BSGS::Logarithm(q);
if(lq%gcd(P-1,p)!=0) {printf("-1\n"); return 0;}
lx = CongruenceEquation(p,lq,P-1);
llong ans = quickpow(G,lx);
printf("%I64d\n",ans);
return 0;
}Codeforces 1106F Lunar New Year and a Recursive Sequence (數學、線性代數、線性遞推、數論、BSGS、擴展歐幾裏得算法)