2019牛客暑期多校训练（第二场） B-Eddy Walker 2

最新推荐文章于 2020-08-11 10:10:15 发布

原创最新推荐文章于 2020-08-11 10:10:15 发布 · 297 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#ACM #杜教BM

数论专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种使用线性代数方法求解在数轴上行走到达特定点的概率问题。通过构建并求解线性递推关系，文章提供了一个高效的算法实现，特别适用于处理有多种步长选择的情况。

Eddy Walker 2

原题（偷懒，截图）

在这里插入图片描述

------------------------------------------------------------------分割线---------------------------------------------------------

题意

一个人从0点出发，在数轴上向右走，
每次有1/K的概率向右走1步，有1/K的概率向右走2步，…，有1/K的概率向右走K步
问到达Ni点的概率是多少
注：Ni==-1时，代表Ni在无穷远处

思路

杜教BM

坑点

板子不能抄错

--------------------------------------------------------分割线-------------------------------------------------------------------

代码

#include <iostream>
#include <cstdio>
#include <vector>
#include <cassert>

using namespace std;

typedef long long ll;

const ll mod=1000000007;
const int N=1e5+5;

ll powmod(ll a,ll b)
{
    ll res=1;
    a%=mod;
    assert(b>=0);
    for(; b; b>>=1)
    {
        if(b&1)
            res=res*a%mod;
        a=a*a%mod;
    }
    return res;
}

ll inv(ll x){return powmod(x,mod-2);}

ll _,n;

namespace linear_seq
{
const int N=10010;
ll res[N],base[N],_c[N],_md[N];

vector<ll> Md;
void mul(ll *a,ll *b,int k)
{
    for(int i=0; i<k+k; i++)
        _c[i]=0;
    for(int i=0; i<k; i++)
        if (a[i])
            for(int j=0; j<k; j++)
                _c[i+j]=(_c[i+j]+a[i]*b[j])%mod;
    for(int i=k+k-1; i>=k; i--)
        if (_c[i])
            for(int j=0; j<(int)Md.size(); j++)
                _c[i-k+Md[j]]=(_c[i-k+Md[j]]-_c[i]*_md[Md[j]])%mod;
    for(int i=0; i<k; i++)
        a[i]=_c[i];
}
ll solve(ll n,vector<ll> a,vector<ll> b)   // a 系数 b 初值 b[n+1]=a[0]*b[n]+...
{
    ll ans=0,pnt=0;
    ll k=(int)(a).size();
    assert((int)a.size()==(int)b.size());
    for(int i=0; i<k; i++)
        _md[k-1-i]=-a[i];
    _md[k]=1;
    Md.clear();
    for(int i=0; i<k; i++)
        if (_md[i]!=0)
            Md.push_back(i);
    for(int i=0; i<k; i++)
        res[i]=base[i]=0;
    res[0]=1;
    while ((1ll<<pnt)<=n)
        pnt++;
    for (int p=pnt; p>=0; p--)
    {
        mul(res,res,k);
        if ((n>>p)&1)
        {
            for (int i=k-1; i>=0; i--)
                res[i+1]=res[i];
            res[0]=0;
            for(int j=0; j<(int)Md.size(); j++)
                res[Md[j]]=(res[Md[j]]-res[k]*_md[Md[j]])%mod;
        }
    }
    for(int i=0; i<k; i++)
        ans=(ans+res[i]*b[i])%mod;
    if (ans<0)
        ans+=mod;
    return ans;
}
vector<ll> BM(vector<ll> s)
{
    vector<ll> C(1,1),B(1,1);
    ll L=0,m=1,b=1;
    for(int n=0; n<(int)s.size(); n++)
    {
        ll d=0;
        for(int i=0; i<=L; i++)
            d=(d+(ll)C[i]*s[n-i])%mod;
        if (d==0)
            ++m;
        else if (2*L<=n)
        {
            vector<ll> T=C;
            ll c=mod-d*powmod(b,mod-2)%mod;
            while ((int)C.size()<(int)B.size()+m)
                C.push_back(0);
            for(int i=0; i<(int)B.size(); i++)
                C[i+m]=(C[i+m]+c*B[i])%mod;
            L=n+1-L;
            B=T;
            b=d;
            m=1;
        }
        else
        {
            ll c=mod-d*powmod(b,mod-2)%mod;
            while ((int)C.size()<(int)B.size()+m)
                C.push_back(0);
            for(int i=0; i<(int)B.size(); i++)
                C[i+m]=(C[i+m]+c*B[i])%mod;
            ++m;
        }
    }
    return C;
}
ll gao(vector<ll> a,ll n)
{
    vector<ll> c=BM(a);
    c.erase(c.begin());
    for(int i=0; i<(int)c.size(); i++)
        c[i]=(mod-c[i])%mod;
    return solve(n,c,vector<ll>(a.begin(),a.begin()+(int)c.size()));
}
};
vector<ll>ans;
ll dp[2*N],p;
int t,k;
int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%lld",&k,&n);
        if(n==-1)
        {
            printf("%lld\n",2*inv(k+1)%mod);//2/(k-1)
            continue;
        }
        p=inv(k);
        dp[0]=1;//dp[n]=1/k dp[n-1] +1/k dp[n-2]+...+1/k dp[n-k]
        ans.clear();
        ans.push_back(dp[0]);
        for(int i=1; i<=2*k; ++i)
        {
            dp[i]=0;
            for(int j=max(i-k,0); j<i; ++j)
            {
                dp[i]=(dp[i]+dp[j])%mod;
            }
            dp[i]=dp[i]*p%mod;//1/k
            ans.push_back(dp[i]);
        }
        printf("%d\n",linear_seq::gao(ans,n));
    }
    return 0;
}