51nod 1149 Pi的递推式 数学

题意

F(x) = 1 (0 <= x < 4)
F(x) = F(x - 1) + F(x - pi) (4 <= x)
Pi = 3.1415926535…..
现在给出一个N，求F(N)。由于结果巨大，只输出Mod 10^9 + 7的结果即可。
1 <= N <= 10^6

分析

题目其实可以看成从0开始，每次往后走1或Pi，问走到(n-4,n]的方案数。
考虑最后一步走1，那就是走到(n-4-1,n-4]的方案数，可以枚举走Pi的次数然后直接组合数算。
最后一步走Pi，那就是走到(n-4-Pi,n-4]的方案数，仍然可以枚举走Pi的次数，但走1的步数需要枚举。枚举的数量不超过4。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#define pi acos(-1)
using namespace std;

typedef long long LL;

const int N=1000005;
const int MOD=1000000007;

int jc[N],ny[N],n;

int C(int n,int m)
{
    return (LL)jc[n]*ny[m]%MOD*ny[n-m]%MOD;
}

int main()
{
    scanf("%d",&n);
    if (n<4) {puts("1");return 0;}
    jc[0]=ny[0]=jc[1]=ny[1]=1;
    for (int i=2;i<=n;i++) jc[i]=(LL)jc[i-1]*i%MOD,ny[i]=(LL)(MOD-MOD/i)*ny[MOD%i]%MOD;
    for (int i=2;i<=n;i++) ny[i]=(LL)ny[i-1]*ny[i]%MOD;
    int ans=0;n-=4;
    for (int i=0;pi*i<=n;i++)
    {
        int x=floor(n-pi*i);
        ans+=C(x+i,x);ans-=ans>=MOD?MOD:0;
    }
    for (int i=0;pi*i<=n;i++)
    {
        int x=floor(n-pi*i);
        while (pi*i+x>n-pi) ans+=C(x+i,x),ans-=ans>=MOD?MOD:0,x--;
    }
    printf("%d",ans);
    return 0;
}