【BZOJ5015】[Snoi2017]礼物 矩阵乘法
【BZOJ5015】[Snoi2017]礼物
Description
热情好客的请森林中的朋友们吃饭,他的朋友被编号为 1~N,每个到来的朋友都会带给他一些礼物:。其中,第一个朋友会带给他 1 个,之后,每一个朋友到来以后,都会带给他之前所有人带来的礼物个数再加他的编号的 K 次方那么多个。所以,假设 K=2,前几位朋友带来的礼物个数分别是:1,5,15,37,83假设 K=3,前几位朋友带来的礼物个数分别是:1,9,37,111现在,好奇自己到底能收到第 N 个朋友多少礼物,因此拜托于你了。已知 N,K请输出第 N 个朋友送的礼物个数 mod1000000007。
Input
第一行,两个整数 N,K
N≤10^18,K≤10
Output
一个整数,表示第 N 个朋友送的礼物个数 mod1000000007。
Sample Input
4 2
Sample Output
37
题解:一开始想到$1^d+2^d+...+n^d$是一个d+1次的多项式,所以猜想它的前缀和也是一个多项式,后来实验了一下死活试不出来。于是换个思路想矩乘,倒是一下就想出来了。。。
我们用S[i]表示$\sum\limits_{i=1}^n\sum\limits_{j=1}^ij^k=\sum\limits_{i=1}^ni^k(n-i+1)$,那么ans=S[n]-S[n-1],所以维护以下行向量:
$\begin{pmatrix}i^0 & i^1 & ... & i^k & S[i] $
如何得到i^k呢?用$(i-1)^0,(i-1)^1,...(i-1)^k$乘上组合数即可。如何得到S[i]呢?用$2*S[i-1]+i^k$即可。然后就没了。
#include <cstdio> #include <cstring> #include <iostream> using namespace std; typedef long long ll; const ll P=1000000007; ll n; ll c[20][20]; int m; struct M { ll a[20][20]; M () {memset(a,0,sizeof(a));} ll * operator [] (int b){return a[b];} M operator * (M b) { M c; int i,j,k; for(i=0;i<=m+1;i++) for(j=0;j<=m+1;j++) for(k=0;k<=m+1;k++) c[i][j]=(c[i][j]+a[i][k]*b[k][j])%P; return c; } }tr,a1,a2; M pm(M ret,ll y) { M x=tr; while(y) { if(y&1) ret=ret*x; x=x*x,y>>=1; } return ret; } int main() { scanf("%lld%d",&n,&m); int i,j; c[0][0]=1; for(i=1;i<=m;i++) { c[i][0]=1; for(j=1;j<=i;j++) c[i][j]=(c[i-1][j-1]+c[i-1][j])%P; } for(i=0;i<=m;i++) { a1[0][i]=a2[0][i]=1; for(j=0;j<=i;j++) tr[j][i]=c[i][j]; } tr[m+1][m+1]=2,tr[m][m+1]=1; a1=pm(a1,n-1),a2=pm(a2,n); printf("%lld",(a2[0][m+1]-a1[0][m+1]+P)%P); return 0; }
| 欢迎来原网站坐坐! >原文链接<