链接:戳这里
思路:
机器人可以选择向右向左或者不走,我们抽象出来一下
机器人不能走到原点的左边,定义机器人当前向右走的步数numR,机器人向左走的步数为numL
那么不动的步数为n-numR-numL
分析一下只有当numR==numL的时候机器人才能回到原点,这个也就是卡他兰数的的经典例子了...略
所以我们所要的方案就是在n步中选出2*i步来让机器人满足 numR=i && numL=i 使得回到原点
枚举这个i并累加贡献
Ans(n)=∑i=0⌊2n⌋Cn2i * Catalan(i)
卡特兰数定义:Catalan(n)= C(2*n,n)/(n+1)
由于这道题需要取模,我这里用到了费马小定理推出
a^b%mod = a^( (b%(mod-1) )%mod
(1/x) % mod = x^(mod-2)%mod
预处理出1e6中的卡特兰数和C(n,2*i)
详细的过程看代码吧
代码:
#include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<string> #include<vector> #include <ctime> #include<queue> #include<set> #include<map> #include<stack> #include<iomanip> #include<cmath> #define mst(ss,b) memset((ss),(b),sizeof(ss)) #define maxn 0x3f3f3f3f #define MAX 1000100 ///#pragma comment(linker, "/STACK:102400000,102400000") typedef long long ll; typedef unsigned long long ull; #define INF (1ll<<60)-1 #define mod 1000000007 using namespace std; ll F[1000100]; ll anw[1000100],cata[1000100],Q[1000100]; ll Qpow(ll a,ll b,ll P){ ll ans=1; while(b){ if(b%2) ans=ans*a%P; a=a*a%P; b/=2; } return ans%P; } ll C(ll n,ll m,ll P){ return F[n]*Q[m]%P*Q[n-m]%P; /* (m!)^(-1)%P = (m!)^(P-2)%P 对于每次的C(n,m,P)来说都需要Qpow一次,强行加了logN 我也是这里一直T 所以预处理出所有的(m!)^(P-2)%P的值省去logN */ } int main(){ Q[0]=F[0]=cata[0]=1; for(int i=1;i<=1000000;i++) { F[i]=F[i-1]*i%mod; Q[i]=Qpow(F[i],mod-2,mod); ///cata[i]=cata[i-1]*(4*i-2)%mod*Qpow(i+1,mod-2,mod)%mod; } /// 预处理出所有的卡特兰数 for(int i=1;i<=1000000;i++) cata[i]=C(2*i,i,mod)*Qpow(i+1,mod-2,mod)%mod; int T,n; scanf("%d",&T); while(T--){ scanf("%d",&n); ll ans=0; for(int i=0;i<=n/2;i++){ ans+=C(n,2*i,mod)*cata[i]%mod; ans%=mod; } printf("%I64d\n",ans); } return 0; }<strong> </strong>