求逆元 - HNU 13412 Cookie Counter

Cookie Counter #

Problem's Link:  http://acm.hnu.cn/online/?action=problem&type=show&id=13412&courseid=0
#

---

 

Mean: 

将N分为D份，每份不超过X，有多少种分法？

analyse:

首先我们想到的是迭代，但是数据太大，一路迭代下去必定爆栈+超内存+TLE。

我们枚举X，对于满足条件的X，求和统计答案，不满足条件的X，更新往下迭代的P值。最后对P求和即为答案。

这题DP也可以做，不过上面的方法从时间和空间上都大大优于DP。

Time complexity: O(N)

 

Source code: 

/*
* this code is made by crazyacking
* Verdict: Accepted
* Submission Date: 2015-08-16-16.39
* Time: 0MS
* Memory: 137KB
*/
#include <queue>
#include <cstdio>
#include <set>
#include <string>
#include <stack>
#include <cmath>
#include <climits>
#include <map>
#include <cstdlib>
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
#define  LL long long
#define  ULL unsigned long long
using namespace std;
const LL mod = 1000000007;
LL inv[5000];
LL N,X,D;
void pre()
{
     inv[1] = 1;
     for(int i=2; i<5000; i++)
           inv[i] = (mod - mod / i) * inv[mod % i] % mod;
}
int main()
{
     pre();
     while(scanf("%d %lld %d",&N,&D,&X) && N)
     {
           LL ans = 0;
           for(int i=0; i*X<=N; i++)
           {
                 LL p = 1;
                 if(i <= D)
                 {
                       for(int j=1; j<=i; j++)
                       {
                             p = (D - j + 1) % mod * p % mod;
                             p = p * inv[j] % mod;
                       }
                 }
                 else p = 0;
                 for(int j=0; j<i; j++) p = (mod - p);
                 int gap = N - i*X;
                 for(int j=1; j<=gap; j++)
                 {
                       p = (D + gap - j + mod) % mod * p % mod;
                       p = p * inv[j] % mod;
                 }
                 ans = ans + p;
                 if(ans >= mod)
                       ans -= mod;
           }
           printf("%lld\n",ans);
     }
     return 0;
}