FZU 2238 - Daxia & Wzc's problem（Lucas）

Problem 2238 Daxia & Wzc's problem

Accept: 42    Submit: 166
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.

 Source

FOJ有奖月赛-2016年8月（daxia专场之过四题方有奖）

POINT：

用a，d表达多写几项就知道是杨辉三角，也就是组合数。答案为：因为是大组合数，用lucas定理

C(i+m -1 ,m)*a+C(i+m -1 , m+ 1 )*d

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define LL long long
const LL mod = 1000000007;
LL quick_mod(LL a,LL b)
{
    LL ans=1;
    a%=mod;
    while(b)
    {
        if(b&1)
        {
            ans=ans*a%mod;
            b--;
        }
        b>>=1;
        a=a*a%mod;
    }
    return ans;
}
LL C(LL n,LL m)
{
    if(m>n) return 0;
    LL ans=1;
    for(int i=1;i<=m;i++)
    {
        LL a=(n+i-m)%mod;
        LL b=i%mod;
        ans=ans*(a*quick_mod(b,mod-2)%mod)%mod;
    }
    return ans;
}
LL Lucas(LL n,LL m)
{
    if(m==0) return 1;
    return C(n%mod,m%mod)*Lucas(n/mod,m/mod)%mod;
}


int main()
{
    int a,d,m,i;
    while(~scanf("%d%d%d%d",&a,&d,&m,&i)){

        printf("%lld\n",(Lucas(i+m-1,m)%mod*a%mod+Lucas(i+m-1, m+1)%mod*d%mod)%mod);
    }
    return 0;
}