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}） ext%p ]

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

分析：

数论板子大集合

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

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

但是这个式子的(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，因为此时有(xqequiv 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
*/