礼物 陕西OI2017 (矩阵快速幂)

 

题目描述

热情好客的小猴请森林中的朋友们吃饭，他的朋友被编号为 1∼N，每个到来的朋友都会带给他一些礼物：香蕉。其中，第一个朋友会带给他1个香蕉，之后，每一个朋友到来以后，都会带给他之前所有人带来的礼物个数再加他的编号的K次方那么多个。所以，假设 K=2，前几位朋友带来的礼物个数分别是：
1,5,15,37,83,…
假设 K=3，前几位朋友带来的礼物个数分别是：
1,9,37,111,…
现在，小猴好奇自己到底能收到第 N 个朋友多少礼物，因此拜托于你了。
已知 N,K，请输出第 N 个朋友送的礼物个数 mod 1000000007。

输入

第一行，两个整数 N,K。

输出

一个整数，表示第N个朋友送的礼物个数 mod 1000000007。

样例输入

4 2

样例输出

37

提示

100% 的数据：N≤1018，K≤10。

来源

陕西OI2017

首先来说下做完这道题目的收获，不仅是只是方面的，发现了自己对一些细节方面的处理还是不是很到位，并且还经常手残敲错很多字母，导致程序运行结果不对(#x心碎#)。

 

拒绝手残和脑残，从现在做起！

再来说下这道题目的思路：   设前n个朋友带的香蕉的总个数为Sn，则Sn = 2*Sn-1 + n^k;设an为第n个朋友带的香蕉的个数，则an = Sn-1 + n^k， n^k = (n-1+1)^k，可以用二项式定理展开。

令 Xn-1 = {Sn-1,C(k,0),C(k,1)......C(k,k)}T, 设系数矩阵为A，

A*Xn-1 = Xn，因为我们要求的是an，所以只要求出Sn-1然后再加上n^k的值就可以啦， 所以也就是先求出Xn-1，A^(n-2)*X1 = Xn-1;

注意下系数矩阵的初始化操作。

代码如下：

 

#include <cstdio>
#include <cstring>
#include <algorithm>
#define ll long long
using namespace std;

const ll mod = 1e9 + 7;
struct Matrix{
	ll a[15][15];
}ori,res;
ll n,k,c[15][15];

void Init(){//初始化操作 
	memset(c,0,sizeof(c));
	c[0][0] = 1;
	for(int i = 1; i <= k; i ++){
		for(int j = 0; j <= k; j ++){
			if(j == 0 || j == i) c[i][j] = 1;
			else c[i][j] = (c[i-1][j-1] + c[i-1][j]) % mod;
		}
	}
	memset(res.a,0,sizeof(res.a));
	for(int i = 0; i <= k+1; i ++)
		res.a[i][i] = 1;
	memset(ori.a,0,sizeof(ori.a));
	ori.a[0][0] = 2;
	for(int i = 1; i <= k+1; i ++)
		ori.a[0][i] = c[k][i-1];
	for(int i = 1; i <= k + 1; i ++){
		for(int j = i, x = 0;  j <= k+1; j ++, x ++){
			ori.a[i][j] = c[k-i+1][x];
		}
	}	
}

Matrix multiply(Matrix x, Matrix y){//矩阵乘法 
	Matrix temp;
	memset(temp.a,0,sizeof(temp.a));
	for(int i = 0; i <= k+1; i ++){
		for(int j = 0; j <= k+1; j ++){
			for(int z = 0; z <= k+1; z ++)
				temp.a[i][j] = (temp.a[i][j] + x.a[i][z] * y.a[z][j]) % mod;
		}
	}
	return temp;
}

ll qpow(ll t, ll k){//快速幂求t^k 
	ll ans = 1;
	t %= mod; //防止相乘后超过long long 的取值范围 
	while(k){
		if(k & 1)  ans = (ans * t) % mod; 
		k >>= 1;
		t = (t * t) % mod;
	}
	return ans;
}

void calcu(ll n){
	ll t = n+2;
	while(n){//矩阵快速幂 
		if(n & 1) res = multiply(res,ori);
		n >>= 1;
		ori = multiply(ori,ori);
	}
	ll sum = 0;   
	for(int i = 0; i <= k+1; i ++)
		sum = (sum + res.a[0][i]) % mod;
	ll ans = qpow(t,k);
	printf("%lld\n",(sum + ans) % mod);
}

int main(){
	while(~scanf("%lld%lld",&n,&k)){
		Init();
		if(n == 1) printf("1\n");
		else calcu(n-2);
	}
	return 0;
}