首页 > 代码库 > 51Nod 1228 -- 伯努利数

51Nod 1228 -- 伯努利数

2024-09-08 06:09:08   214人阅读

题目大意：

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的结果即可。
其中1<=T<=5000,1<=n<=108,1<=k<=2000
 
可以用伯努利数求自然数k次幂和。公式：
 
技术分享
 
其中B表示伯努利数。
那么我们可以预处理出B,C，然后就可以O(k)计算答案了。
公式：
技术分享

技术分享

 总时间复杂度O(T*k)

 

代码：

技术分享 
 1 #include<cstdio>
 2 #include<cstring>
 3 #include<iostream>
 4 #include<cmath>
 5 using namespace std;
 6 #define M 1000000007
 7 #define K 2010
 8 #define ll long long
 9 int T,i;
10 ll k,m,inv[K],n,j,c[K][K],b[K],Ans;
11 int main()
12 {
13     for(inv[1]=1,i=2;i<K;i++)inv[i]=(M-M/i)*inv[M%i]%M;
14     for(i=1;i<K;i++){
15         c[i][0]=c[i][i]=1;
16         for(j=1;j<i;j++)
17         c[i][j]=(c[i-1][j]+c[i-1][j-1])%M;
18     }
19     for(b[0]=1,i=1;i<K-1;i++){
20         for(j=0;j<i;j++)b[i]=(b[i]-b[j]*c[i+1][j])%M;
21         b[i]=(b[i]*inv[i+1])%M;
22     }
23     scanf("%d",&T);
24     while(T--){
25         scanf("%I64d%d",&n,&k);
26         Ans=0;n%=M;
27         for(i=1,j=n+1;i<=k+1;i++,j=(j*(n+1))%M)Ans=(Ans+(c[k+1][i]*b[k-i+1]%M)*j%M)%M;
28         Ans=(Ans*inv[k+1])%M;
29         printf("%I64d\n",(Ans+M)%M);
30     }
31     return 0;
32 }
51Nod 1228

 

51Nod 1228 -- 伯努利数


联系
我们