CF1106F Lunar New Year and a Recursive Sequence

时间:2023-03-10 07:25:18
CF1106F Lunar New Year and a Recursive Sequence

题目链接:CF1106F Lunar New Year and a Recursive Sequence

大意:已知\(f_1,f_2,\cdots,f_{k-1}\)和\(b_1,b_2,\cdots,b_k\),且有递推关系

\[f_i=(\prod_{j=1}^kf_{i-j}^{b_j})\text%p
\]

对于所有\(i>k\)均成立,给出\(f_n=m\),求\(f_k\),\(p=998244353\)

分析:

数论板子大集合

首先很显然有\(f_n=f_k^{q}\),于是考虑先求出这个\(q\)

看到\(k\leq100\)一般就会想到矩阵乘法之类的

但是这个式子的\(b\)处在乘方的位置,无法直接使用矩阵

注意到模数为998244353,其原根为3

那么我们就可以使用原根来改写这个转移

\[g_i=\prod_{j=1}^kg_{i-j}b_j
\]

这个是可以利用矩阵乘法的,初值\(g_i=0(i<k),g_k=1\)

我们于是可以利用矩乘找到这样的关系式:\(f_k^{q}\equiv f_n(mod\ p),q=g_n\)

这个玩意似乎可以类比的被定义为k次剩余,不过我们显然不会这个玩意

于是我们继续使用原根

我们知道\(f_n\)可以被写作\(3^t\),那么我们设\(f_k\)可以被写作为\(3^x\)

那么我们就是求满足这个式子的x值:\(3^{xq}\equiv3^t(mod\ p)\)

那么对于指数我们可以直接用exgcd,因为此时有\(xq\equiv t(mod\ p-1)\)

将m转为\(3^t\)时可以使用BSGS

问题就得到了解决QAQ

注意模数会在\(p-1\)和\(p\)之间切换,指数的模数为\(p-1\)(包括上面的矩乘),而在BSGS中的模数则为\(p\)

#include<iostream>
#include<string.h>
#include<string>
#include<stdio.h>
#include<algorithm>
#include<math.h>
#include<vector>
#include<queue>
#include<map>
using namespace std;
#define int long long
const int maxd=998244353,N=100000;
const double pi=acos(-1.0);
typedef long long ll;
int n,K,b[120],m; struct matrix{
ll x[120][120];
}ans,sum; matrix operator *(matrix a,matrix b)
{
matrix c;
memset(c.x,0,sizeof(c.x));
int i,j,k;
for (i=0;i<K;i++)
{
for (j=0;j<K;j++)
{
for (k=0;k<K;k++)
{
c.x[i][j]=(c.x[i][j]+a.x[i][k]*b.x[k][j])%(maxd-1);
}
}
}
return c;
} void qpow(int tim)
{
while (tim)
{
int tmp=tim&1;tim/=2;
if (tmp) ans=ans*sum;
sum=sum*sum;
}
} ll qpow(ll x,ll y)
{
ll ans=1,sum=x;
while (y)
{
int tmp=y&1;y/=2;
if (tmp) ans=(ans*sum)%maxd;
sum=(sum*sum)%maxd;
}
return ans;
} map<ll,ll> mp;
ll bsgs(int a,int b)
{
int siz=sqrt(maxd-1)+1,i;ll sum=b;
for (i=0;i<=siz;i++)
{
mp[sum]=i;
sum=(sum*a)%maxd;
}
ll tmp=qpow(a,siz);sum=1;
for (i=1;i<=siz;i++)
{
sum=(sum*tmp)%maxd;
if (mp[sum]) return i*siz-mp[sum];
}
} ll gcd(ll x,ll y)
{
if (!y) return x; else return gcd(y,x%y);
} void exgcd(ll a,ll b,ll &g,ll &x,ll &y)
{
if (!b) {x=1;y=0;g=a;return;}
else
{
ll tmpx,tmpy;
exgcd(b,a%b,g,tmpx,tmpy);
x=tmpy;y=tmpx-a/b*tmpy;
}
} int read()
{
int x=0,f=1;char ch=getchar();
while ((ch<'0') || (ch>'9')) {if (ch=='-') f=-1;ch=getchar();}
while ((ch>='0') && (ch<='9')) {x=x*10+(ch-'0');ch=getchar();}
return x*f;
} signed main()
{
K=read();
int i,j;
for (i=0;i<K;i++) {ans.x[i][i]=1;sum.x[0][i]=read();}
for (i=0;i<K-1;i++) sum.x[i+1][i]=1;
ans.x[0][0]=1;
n=read();m=read();
qpow(n-K);
ll q=ans.x[0][0],p=bsgs(3,m);
//cout << q << " " << p << endl;
ll tmp=gcd(q,maxd-1);
if (p%tmp) {printf("-1");return 0;}
ll x,y,g;
exgcd(q,maxd-1,g,x,y);
x=(p/g*x)%(maxd-1);
if (x<0) x+=(maxd-1);
printf("%lld",qpow(3,x));
return 0;
}
/*
3
2 3 5
4 16
*/