51nod 1228 序列求和（伯努利数)

1228 序列求和 

题目来源： HackerRank

基准时间限制：3 秒 空间限制：131072 KB 分值: 160 难度：6级算法题

 收藏

 关注

T(n) = n^k，S(n) = T(1) + T(2) + ...... T(n)。给出n和k，求S(n)。

例如k = 2，n = 5，S(n) = 1^2 + 2^2 + 3^2 + 4^2 + 5^2 = 55。

由于结果很大，输出S(n) Mod 1000000007的结果即可。

Input

第1行：一个数T，表示后面用作输入测试的数的数量。（1 <= T <= 5000)
第2 - T + 1行：每行2个数，N, K中间用空格分割。(1 <= N <= 10^18, 1 <= K <= 2000)

Output

共T行，对应S(n) Mod 1000000007的结果。

Input示例

3
5 3
4 2
4 1

Output示例

225
30
10

#include<bits/stdc++.h>

#define MOD 1000000007

#define P 1000000007

using namespace std;

typedef long long ll;

int T;ll n;int k;

int fac[2005],inv[2005];

int c[2005][2005];

int B[2005];//伯努利数

void init()//伯努利数打表

{

    //Pre(2004);

    B[0]=1;

c[0][0]=1;

    for (int i=1;i<2004;i++)

    {

        for (int j=1;j<=i;j++) c[i][j]=(c[i-1][j-1]+c[i-1][j]) % MOD;

        c[i][0]=1;

    }

     inv[1]=1;

    for (int i=2;i<2001;i++) inv[i]=(ll)inv[MOD % i] * (MOD-MOD/i) % MOD;

    for (int i=1;i<2001;i++)

    {

        B[i]=0;

        for (int k=0;k<i;k++) B[i]=(B[i]+(ll)c[i+1][k]*B[k] % MOD) % MOD;

        B[i]=((ll)B[i]*(-inv[i+1]) % MOD+MOD)%MOD;

    }

}

ll ans;

int main()

{

#ifndef ONLINE_JUDGE

    freopen("in.txt","r",stdin);

#endif // ONLINE_JUDGE

    init();

    ll tmp;

   scanf("%d",&T);

    while (T--)

    {

        cin>>n>>k;

        n++; n%=MOD; tmp=n;

        ans=0;

        for (int i=1;i<=k+1;i++)

        {

            ans=(ans+(ll)c[k+1][i]*B[k+1-i]%MOD*n%MOD) % MOD;

            n=(ll)n*tmp % MOD;

        }

        ans=(ll)ans*inv[k+1] % MOD;

        cout<<ans<<endl;

    }

return 0;

}