CF 717A Festival Organization——斯特林数+递推求通项+扩域

题目:http://codeforces.com/contest/717/problem/A

是 BJOI2019 勘破神机 的弱化版。

令 \( g[i] \) 表示长为 i 、以 1 结尾的方案数,有 \( g[i]=g[i-1]+g[i-2] , g[0]=g[1]=1 \) ;

令 \( f[i] \) 表示长为 i 的方案数,有 \( f[i]=g[i]+g[i-1] \)

发现 \( f[i]=f[i-1]+f[i-2] , f[0]=1 , f[1]=2 \)

那么令 l+=2 , r+=2 , f[ i ] 就是普通的斐波那契数。用 BJOI 那道题的套路即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=205,mod=1e9+7;
int upt(int x){while(x>=mod)x-=mod;while(x<0)x+=mod;return x;}
int pw(int x,int k)
{int ret=1;while(k){if(k&1)ret=(ll)ret*x%mod;x=(ll)x*x%mod;k>>=1;}return ret;}

int k,s[N][N],c[N][N],ans;
int tlen;ll l,r,len;
struct Node{
  int x,y;
  Node(int x=0,int y=0):x(x),y(y) {}
  Node operator+ (const Node &b)const
  {return Node(upt(x+b.x),upt(y+b.y));}
  Node operator- (const Node &b)const
  {return Node(upt(x-b.x),upt(y-b.y));}
  Node operator* (const Node &b)const
  {return Node(((ll)x*b.x+(ll)y*b.y%mod*5)%mod,((ll)x*b.y+(ll)y*b.x)%mod);}
}A[N],B[N],x1[N],x2[N],one;
Node pw(Node x,ll k)
{Node ret=one;while(k){if(k&1)ret=ret*x;x=x*x;k>>=1;}return ret;}
void init()
{
  s[0][0]=1;
  for(int i=1;i<=k;i++)
    for(int j=1;j<=i;j++)
      s[i][j]=(s[i-1][j-1]+(ll)s[i-1][j]*(i-1))%mod;
  for(int i=0;i<=k;i++)c[i][0]=1;
  for(int i=1;i<=k;i++)
    for(int j=1;j<=i;j++)
      c[i][j]=upt(c[i-1][j-1]+c[i-1][j]);

  one=Node(1,0); int tp=pw(5,mod-2);
  A[1]=Node(0,tp); B[1]=Node(0,upt(-tp));
  tp=pw(2,mod-2); x1[1]=Node(tp,tp); x2[1]=Node(tp,upt(-tp));
  A[0]=B[0]=x1[0]=x2[0]=one;
  for(int i=2;i<=k;i++)A[i]=A[i-1]*A[1];
  for(int i=2;i<=k;i++)B[i]=B[i-1]*B[1];
  for(int i=2;i<=k;i++)x1[i]=x1[i-1]*x1[1];
  for(int i=2;i<=k;i++)x2[i]=x2[i-1]*x2[1];
}
Node Inv(Node x)
{
  int tp=upt(((ll)x.x*x.x-(ll)x.y*x.y%mod*5)%mod);
  tp=pw(tp,mod-2);
  return Node((ll)x.x*tp%mod,upt(-(ll)x.y*tp%mod));
}
Node cal(Node x)
{
  if(x.x==1&&x.y==0)return Node(tlen,0);
  return pw(x,l)*(one-pw(x,len))*Inv(one-x);
}
int main()
{
  scanf("%d%lld%lld",&k,&l,&r); l+=2; r+=2;
  init(); len=r-l+1; tlen=len%mod;
  for(int j=0,fx=((k&1)?upt(-1):1);j<=k;j++,fx=upt(-fx))
    {
      int tp=0;
      for(int t=0;t<=j;t++)
    {
      Node d=cal(x1[t]*x2[j-t]);
      d=d*A[t]*B[j-t];
      tp=(tp+(ll)c[j][t]*d.x)%mod;
    }
      ans=(ans+(ll)s[k][j]*fx%mod*tp)%mod;
    }
  int ml=1;
  for(int i=2;i<=k;i++)ml=(ll)ml*i%mod;
  ml=pw(ml,mod-2);
  ans=(ll)ans*ml%mod;
  printf("%d\n",ans);
  return 0;
}

 

posted on 2019-05-29 22:20  Narh  阅读(232)  评论(0编辑  收藏  举报

导航