牛客小白月赛6 J 洋灰三角

题目描述

    洋灰是一种建筑材料，常用来筑桥搭建高层建筑，又称，水泥、混凝土。

    WHZ有很多铸造成三角形的洋灰块，他想把这些洋灰三角按照一定的规律放到摆成一排的n个格子里，其中第i个格子放入的洋灰三角数量是前一个格子的k倍再多p个，特殊地，第一个格子里放1个。
    WHZ想知道把这n个格子铺满需要多少洋灰三角。

输入描述:

第一行有3个正整数n，k，p。

输出描述:

输出一行，一个正整数，表示按照要求铺满n个格子需要多少洋灰三角，由于输出数据过大，你只需要输出答案模1000000007(1e9+7)后的结果即可。

 

示例1

输入

3 1 1

输出

6

说明

洋灰三角铺法：1 2 3，总计6个

示例2

输入

3 2 2

输出

15

说明

洋灰三角铺法：1 4 10，总计15个

示例3

输入

3 3 3

输出

28

说明

洋灰三角铺法：1 6 21，总计28个

备注:

对于100%的测试数据：
1 ≤ n ≤ 1000000000
1 ≤ k,p ≤ 1000

 

数据范围有点吓人。不会处理矩阵的常数项。。。

之前做过一个题，求的是an=m0an−1+m1an−2+c不会构造矩阵，然后题解却是O(1)的快速乘，然后，，然后就没有然后了

这题有几个公式

Sn=Sn-1+an  (an=k*an-1+p)

an=k*an-1+p

p=p

 

 

然后就是矩阵快速幂加速一下就行了

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define ll long long
const int maxn=5;
const long long mod=1e9+7;
int n,k,p;
typedef struct Node{
	ll num[maxn][maxn];
}node;
node E,tt;
void init(){
	
	tt.num[1][1]=tt.num[1][3]=tt.num[3][3]=tt.num[2][3]=1;
	tt.num[1][2]=tt.num[2][2]=k;
}
node mul(Node a,Node b){
	Node c;
	int i,j,k;
	for(i=1;i<=3;i++){
		for(j=1;j<=3;j++){
			c.num[i][j]=0;
			for(k=1;k<=3;k++){
				c.num[i][j]+=a.num[i][k]*b.num[k][j];
				c.num[i][j]%=mod;
			}
		}
	}
	return c;
}
node po(node a,int b){
	node ans;
	for(int i=1;i<=3;i++){
		for(int j=1;j<=3;j++){
			if(i==j) ans.num[i][i]=1;
			else ans.num[i][j]=0;
		}
	}
	while(b){
		if(b&1) ans=mul(ans,a);
		a=mul(a,a);
		b>>=1;
	}
	return ans;
}
int main(){
	int i,j;
	while(scanf("%d%d%d",&n,&k,&p)!=EOF){
		if(n==1){
			printf("1\n");
			continue;
		}
		else if(n==2){
			printf("%lld\n",k+p);
			continue;
		}
		init();
		node ans=po(tt,n-1);
		ll sum=ans.num[1][1]+ans.num[1][2]+(p*ans.num[1][3])%mod;
		sum%=mod;
		printf("%lld\n",sum);
	}
	return 0;
}

 

还有大佬是直接推公式得Sn

还有sn=(k+1)sn-1-ksn-2+q

 

在矩阵中负数要  -k <=> mod-k;