组合数学--51Nod 1149 Pi的递推式

传送门

$F(i)$可以理解成从数轴上位置$i$出发每次向左走$1$或$Pi$个单位，使最终位置刚好小于$4$的方案数，枚举走了多少个$1$，判断一下最后剩下的如果$\ge 3$就可以走$1$，否则只能走$Pi$

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define maxn 1000005
#define LL long long
using namespace std;
int n,fac[maxn],inv[maxn];
LL ans;
const int mod=1e9+7;
const double pi=acos(-1.0);

inline int qpow(int x,int k){
	int ret=1;
	while(k){
		if(k&1) ret=1LL*ret*x%mod;
		x=1LL*x*x%mod; k>>=1;
	} return ret;
}

inline LL C(int n,int m){ return 1LL*fac[n]*inv[m]%mod*inv[n-m]%mod;}

int main(){
	scanf("%d",&n); fac[0]=1;
	if(n<=3) return puts("1"),0;
	for(int i=1;i<=n;i++) fac[i]=1LL*fac[i-1]*i%mod;
	inv[n]=qpow(fac[n],mod-2);
	for(int i=n;i;i--) inv[i-1]=1LL*inv[i]*i%mod;
	for(int i=0;i<=n-3;i++){
		int x=(int)(ceil(1.0*(n-i-4)/pi));
		if(n-i-(pi*x)>=3) (ans+=C(i+x,i))%=mod;
		else (ans+=C(i+x-1,i))%=mod;
	}
	printf("%lld\n",ans);
	return 0;
}