Description
F(x) = 1 (0 <= x < 4)
F(n) = F(n - 1) + F(n - pi) (4 <= n)
Pi = 3.1415926535…..
现在给出一个N,求F(n)。由于结果巨大,只输出Mod 10^9 + 7的结果即可。
1 <= N <= 10^6
Solution
之前在乱搞生成函数的时候突然get到了一个idea(然而和生成函数并没有什么关系)
因为在推Fib的生成函数的时候,突然发现了一个神奇的东西。
好吧,我们先抛开这道题,想想一道经典的题。
上楼梯,你每一次可以上一层楼或两层楼,求上到n层的方案数。
设F(i)表示上到第i层的方案数,显然F(i)=F(i-1)+F(i-2)
咦?这不是就是Fib吗?
也就是说,Fib的第i项还可以表示,一堆1和2相加得到i的方案数!!!
这样子我们就可以枚举1的个数,把Fib的第i项用组合数表示出来!!!
也就是说,我们可以用O(n)的时间用组合数求出Fib的第i项!!!
然而这并没有什么卵用。。。
回到这道题,你发现了什么?
一堆1和pi加起来与n的差<4的方案数!!!
那么分类讨论,枚举最后个是1还是pi,然后剩下的用组合数加起来就好了。
推广一下,也就是说,这一类的F(i)=F(i-a)+F(i-b)的问题,我们都可以O(n)地把答案用组合数表示出来!!!
(然而这并没有什么卵用,不过好像好厉害的样子)
#include <cmath>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
typedef double db;
typedef long long ll;
const int mo=1e9+7,N=1e6+5;
ll fact[N],inv[N],ans;
int n;
ll c(int m,int n) {
return fact[m]*inv[n]%mo*inv[m-n]%mo;
}
ll mi(ll x,int y) {
ll z=1;
for(;y;y/=2,x=x*x%mo) if (y&1) z=z*x%mo;
return z;
}
int main() {
scanf("%d",&n);fact[0]=inv[0]=1;
if (n<4) {printf("1");return 0;}
fo(i,1,n) fact[i]=fact[i-1]*i%mo;
inv[n]=mi(fact[n],mo-2);
fd(i,n-1,1) inv[i]=inv[i+1]*(i+1)%mo;
fo(i,0,n-4) {
int j=(db)(n-4-i)/M_PI;
(ans+=c(j+i,i))%=mo;
}
fo(i,0,(n-4)/M_PI) {
int j=(db)(n-4-i*M_PI);
(ans+=c(j+i,i))%=mo;
}
printf("%lld",ans);
}