杭电1 :Pass!

题面:有n个人每过一秒传一次球,现知道t秒能有x种最后传回第一个人的可能,求t的最小值。
解析:设x=f(t),所以易得图片说明
接着求解特征方程,得到特征根-1,n-1;
图片说明
带入t=0时f(t)=1,和t=1时f(t)=0;
图片说明
之后因为已知f(x)和n,分奇偶考虑,就可以用bsgs求t。

bsgs :形如图片说明 ,若图片说明 ,方程的解就满足0<=x<p。

图片说明 ,其中图片说明
则转化为图片说明
剩下只需要枚举B,用unorder_map存右边,再枚举A,寻找左右是否相等,即求出x。
代码

#include<bits/stdc++.h>
using namespace std;
const int p=998244353;
typedef long long ll;
ll kpow(ll a,ll b){
    ll ans=1;
    while(b){
        if(b&1){
            ans=ans*a%p;
        }
        a=a*a%p;
        b>>=1;
    }
    return ans%p;
}

unordered_map<ll,ll> mp;
ll bsgs(ll a,ll b){
    mp.clear();
    a%=p;b%=p;
    if(b==1) return 0;
    ll A=ceil(sqrt(p));
    for(ll i=0,j=b;i<A;i++)
    {
        mp[j]=i;
        j=a*j%p;
    }
    ll aa=kpow(a,A);
    for(ll i=1,num=aa;i<=A;i++){
        if(mp.count(num))return i*A-mp[num];
        num=num*aa%p;
    }

    return -1;

}
ll t,n,x;
ll a,b;
int main(){
    scanf("%lld",&t);

    while(t--){
        scanf("%lld%lld",&n,&x);

        if(x==1) puts("0");
        else if(x==0) puts("1");//特判
        else{
        a=n-1;
        b=x*n%p-(n-1);
        b=(b+p)%p;//防止b小于0
        ll ans =bsgs(a,b);
        if(ans&1) ans=-1;

        b=x*n%p+(n-1);
        b=(b+p)%p;
        ll res=bsgs(a,b);       
        if(res%2==0) res=-1;

        if(res == ans && res == -1) puts("-1");
        else if(res == -1) printf("%lld\n", ans);
        else if(ans == -1) printf("%lld\n", res);
        else printf("%lld\n", min(res, ans));
    }
    }
    }
全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务