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;
}