题目大意: 有n个人互相传球,一开始球在1号处,每过1轮,一个人可以把球传给除自己以外的其他人,设t轮过后球在1号处,这样的传球方案数%998244353有x种,现在给你n和x,求t
2<=n<=10^6
0<=x<998244353
思路:
设f[i][0]表示经过i轮过后球在1号处,f[i][1]表示球经过i轮后不在1号处
f[i][0]=f[i-1][1]
f[i][1]=f[i-1][0](n-1)+f[i-1][1](n-2)
–> f[i-1][1]=f[i-2][0](n-1)+f[i-2][1](n-2)
–> f[i][0]=f[i-1][1]=f[i-2][0](n-1)+f[i-2][1](n-2)
–> f[i-1][0]=f[i-2][1]
–> f[i][0]=f[i-2][0](n-1)+f[i-1][0](n-2)
可以化简一下,转换成1维的,就成了
f[i]=f[i-2](n-1)+f[i-1](n-2)
当然,可以不这么推理,直接思考,f[i-2](n-1)都很容易理解,就是1传给别人再传回来,f[i-1](n-2)有人觉得不能理解,i-1轮在1,为什么i轮也在1,但是你想想,i-1轮在1,和i-2轮在除1之外的其他位置的方案数一样吧,然后你从i-2轮绕一个人再回到1,绕的这个人是不是只有(n-2)种选法
有了转移方程,我们想求f[t]的通项公式,感觉很熟悉,就是根据特征方程求通项公式(高数似乎学过~)
转换为方程x^2-(n-2)*x-(n-1)=0
两个特征根,x1=-1,x2=n-1
把f(1)=0,f(2)=n-1两个特殊值代入,求得
f(t)=(n-1)/n *(-1)^t+1/n (n-1)^t=x%mod
n放到右边
(n-1) * (-1)^t + (n-1) ^t =nx%mod;
这个(-1)^t就有点烦了,使得我的式子无法化简,然后就把t分奇偶情况考虑,就可以化成A ^x=B%mod的形式了,然后用bsgs模板分别求一下t为奇偶的情况,存在的t取一个最小值即可
代码:
#include<bits/stdc++.h>
#define ll long long
const ll mod=998244353;
using namespace std;
ll qpow(ll a,ll b)
{
ll res = 1;
while(b)
{
if(b&1) res = res*a%mod;
a = a*a%mod;
b >>= 1;
}
return res;
}
unordered_map<int,int>Hash;
ll bsgs(ll a, ll b, ll p) {
Hash.clear();
a %= p; b %= p;
if(b == 1) return 0;
int k = sqrt(p) + 1;
for(int i = 0, j = b % p; i < k; i++) {
Hash[j] = i;
j = (ll)j * a % p;
}
int ak = qpow(a, k);
for(int x = 1, num = ak; x <= k; x++) {
if(Hash.count(num)) return x * k - Hash[num];
num = (ll)num * ak % p;
}
return -1;
}
int main()
{
ll t;
scanf("%d",&t);
while(t--){
ll n,x;
scanf("%lld%lld",&n,&x);
ll p1=(n*x%mod+(n-1)%mod)%mod;
ll p2=(n*x%mod+(1-n+mod)%mod)%mod;
ll k1=bsgs(n-1,p1,mod);
if(k1%2==0) k1=-1;
ll k2=bsgs(n-1,p2,mod);
if(k2%2) k2=-1;
if(k1==-1&&k2==-1) puts("-1");
else if(k1==-1){
printf("%lld\n",k2);
}
else if(k2==-1){
printf("%lld\n",k1);
}
else{
ll res=min(k1,k2);
printf("%lld\n",res);
}
}
return 0;
}