BZOJ5015[Snoi2017]礼物 矩阵乘法

Posted 2020-10-06 CQzhangyu

【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。

PDF题面:www.lydsy.com/JudgeOnline/upload/gift.pdf

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