FZU Problem 2238 Daxia & Wzc's problem(Lucas定理求组合数)

此文章可以使用目录功能哟↑(点击上方[+])

 FZU Problem 2238 Daxia & Wzc's problem

Accept: 0    Submit: 0
Time Limit: 1000 mSec    Memory Limit : 32768 KB

 Problem Description

Daxia在2016年5月期间去瑞士度蜜月,顺便拜访了Wzc,Wzc给他出了一个问题：

Wzc给Daxia等差数列A(0),告诉Daxia首项a和公差d;

首先让Daxia求出数列A(0)前n项和,得到新数列A(1);

然后让Daxia求出数列A(1)前n项和,得到新数列A(2);

接着让Daxia求出数列A(2)前n项和,得到新数列A(3);

...

最后让Daxia求出数列A(m-1)前n项和,得到新数列A(m);

 Input

测试包含多组数据,每组一行,包含四个正整数a(0<=a<=100),d(0<d<=100),m(0<m<=1000),i(1<=i<=1000000000).

 Output

每组数据输出一行整数,数列A(m)的第i项mod1000000007的值.

 Sample Input

1 1 3 4

 Sample Output

35

 Hint

A(0): 1 2 3 4

A(1): 1 3 6 10

A(2): 1 4 10 20

A(3): 1 5 15 35

So the 4th of A(3) is 35.

 Problem Idea

解题思路:

【题意】
等差数列A(0)的首项为a,公差为d

数列A(j)(j>0)的第n项为数列A(j-1)的前n项和

求数列A(m)的第n项mod 1000000007的值为多少

【类型】
Lucas定理

【分析】
拿到此题,显然要先观察数列A(m)到底是个怎么样的数列

上述表是可以初步得到的,而罗马数字标识部分则是需要进一步求解的,特别的,Ⅵ处为我们要求的结果

最初的时候,虽说A(j)的第n项的值为A(j-1)的前n项和,但也可以看成是A(j)的第n-1项+A(j-1)的第n项

即A[i][j]=A[i-1][j]+A[i][j-1]

此时,我最先想到的其实是矩阵构造

但是由于构造的矩阵规模比较大,所以放弃了这种想法

那就只能专心求递推式了

我们一步步来,首先先求Ⅰ处的递推式

因为数列A(0)是等差数列,所以数列A(1)的第n项可以通过等差数列求和公式得到

接着是Ⅱ处的推导

由此可得,Ⅵ处的公式为

在用lucas定理求得组合数之后,剩下的就是代入公式了

【时间复杂度&&优化】

题目链接→FZU Problem 2238 Daxia & Wzc's problem

 Source Code

/*Sherlock and Watson and Adler*/
#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<queue>
#include<stack>
#include<math.h>
#include<vector>
#include<map>
#include<set>
#include<cmath>
#include<complex>
#include<string>
#include<algorithm>
#include<iostream>
#define eps 1e-9
#define LL long long
#define PI acos(-1.0)
#define bitnum(a) __builtin_popcount(a)
using namespace std;
const int N = 1003;
const int M = 40;
const int inf = 1000000007;
const int mod = 1000000007;
__int64 quick_mod(__int64 a,__int64 b)
{
    __int64 ans=1;
    a%=mod;
    while(b)
    {
        if(b&1)
        {
            ans=ans*a%mod;
            b--;
        }
        b>>=1;
        a=a*a%mod;
    }
    return ans;
}
__int64 C(__int64 n,__int64 m)
{
    if(m>n) return 0;
    __int64 ans=1;
    for(int i=1;i<=m;i++)
    {
        __int64 a=(n+i-m)%mod;
        __int64 b=i%mod;
        ans=ans*(a*quick_mod(b,mod-2)%mod)%mod;
    }
    return ans;
}
__int64 Lucas(__int64 n,__int64 m)
{
    if(m==0) return 1;
    return C(n%mod,m%mod)*Lucas(n/mod,m/mod)%mod;
}
int main()
{
    int a,d,m,n;
    while(~scanf("%d%d%d%d",&a,&d,&m,&n))
    {
        printf("%I64d\n",(Lucas(n+m-1,m)*a+Lucas(n+m-1,m+1)*d)%mod);
    }
    return 0;
}

菜鸟成长记