HDU 5136 Yue Fei's Battle

题目链接：HDU-5136

网上的一篇题解非常好，所以就直接转载了。转自oilover的博客

代码：

 #include<cstring>
 #include<cstdio>
 #include<queue>
 using namespace std;
 typedef long long LL;
 const LL MAXN=;
 const LL MOD=;

 LL dp[MAXN+],sum[MAXN+];
 LL mod2,mod6;
 LL extgcd(LL a,LL b,LL &x,LL &y)
 {
     LL d=a;
     if(b!=)
     {
         d=extgcd(b,a%b,y,x);
         y-=(a/b)*x;
     }
     else { x=; y=; }
     return d;
 }
 LL modInverse(LL a,LL m)
 {
     LL x,y;
     extgcd(a,m,x,y);
     return (m+x%m)%m;
 }
 void init()
 {
     mod2=modInverse(,MOD);
     mod6=modInverse(,MOD);
     memset(dp,,sizeof(dp));
     memset(sum,,sizeof(sum));
     dp[]=dp[]=;
     sum[]=;
     sum[]=;
     for(LL i=;i<=MAXN;i++)
     {
         dp[i]=(dp[i-]*sum[i-] % MOD + dp[i-]*(dp[i-]+) %MOD *mod2 % MOD)%MOD;
         sum[i]=(sum[i-]+dp[i])%MOD;
     }
 }
 LL f(LL kk)
 {
     LL k=kk/;
     if(kk%==)
         return dp[k]*(dp[k]+)%MOD *mod2 % MOD;
     LL ans=;
     ans =(ans + dp[k]*(dp[k]+)%MOD *mod2 %MOD *sum[k-] %MOD) %MOD;
     ans =(ans + dp[k]) %MOD;
     ans =(ans + dp[k]*((dp[k]-+MOD)%MOD) %MOD) %MOD;
     if(dp[k]>=) ans =(ans + dp[k]*(dp[k]-)%MOD*(dp[k]-)%MOD*mod6%MOD)%MOD;
     return ans;
 }
 int main()
 {
 #ifdef LOCAL
     freopen("in.txt","r",stdin);
     // freopen("out.txt","w",stdout);
 #endif
     init();
     LL k;
     while(scanf("%lld",&k)!=EOF && k)
         printf("%lld\n",f(k));
     return ;
 }