链接:戳这里
思路:
机器人可以选择向右向左或者不走,我们抽象出来一下
机器人不能走到原点的左边,定义机器人当前向右走的步数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>