【矩阵乘法】【SSL 2514】幼儿园数学题Ⅱ

最新推荐文章于 2024-08-20 07:30:00 发布

ssl_yty

最新推荐文章于 2024-08-20 07:30:00 发布

阅读量98

点赞数

分类专栏：快速幂矩阵乘法

本文链接：https://blog.csdn.net/qq_45621109/article/details/111407172

版权

快速幂同时被 2 个专栏收录

10 篇文章 0 订阅

订阅专栏

矩阵乘法

9 篇文章 0 订阅

订阅专栏

【矩阵乘法】【SSL 2514】幼儿园数学题Ⅱ

题目

这天,当一头雾水的LZH同学在考场上痛哭的时候,一旁的YMW早就如切菜一样cut掉了简单至极的第一题,风轻云淡的冲击着满分,然而最后一道题着实难道了他,毕竟是幼儿园副园长树皮和著名毒瘤秋彪为了防止人AK而出的,可是YMW作为ACrush的著名粉丝,向来以AK为目标,永不言败,而他能不能AK就看你了
题目是酱紫的,f(n)-f(3)-f(4)-f(5)-…-f(n-3)-f(n-2)=(n+4)(n-1)/2,f(1)=1,f(2)=1
求f(n)的前n项和

输入

输入一个正整数n（保证0<=n<=2^31-1）

输出

输出一个正整数,表示这个图形的整点个数,需要对1000000007求余

样例

input1
1

output1
1

input2
2

output 2
2

解题思路

题目给出f[n]=f[3]+f[4]+…f[n-3]+f[n-2]+(n+4)(n-1)
那可以得出f[n+1]=f[3]+f[4]+…f[n+1-3]+f[n+1-2]+(n+1+4)(n+1-1)
可以发现f[n+1]比f[n]多了f[n-1]+n+2
f[n+1]=f[n]+f[n-1]+(n+1)+1
ta就很眼熟啊
把之前的程序加个求和就OK啦

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const long long mo=1000000007;
struct lzf{
	long long n,m,h[10][10];
}a,b,x;
long long n;
lzf operator *(lzf l,lzf y)
{
	x.n=l.n,x.m=y.m;
	memset(x.h,0,sizeof(x.h));
	for (int k=1;k<=l.m;k++)
	    for (int i=1;i<=x.n;i++)
	        for (int j=1;j<=x.m;j++)
	            x.h[i][j]=(x.h[i][j]+l.h[i][k]*y.h[k][j]%mo)%mo;
	return x;
}
void power(long long n)
{ 
     while (n)
     {
	       if (n & 1) a=a*b;
	       b=b*b;
	       n>>=1;
	 }
}
int main()
{
	a.n=1,a.m=5;
	a.h[1][1]=a.h[1][2]=a.h[1][4]=1;
	a.h[1][3]=3,a.h[1][5]=1;
	b.n=5,b.m=5;
	b.h[1][2]=b.h[2][1]=b.h[2][2]=b.h[3][2]=1;
	b.h[3][3]=b.h[4][2]=b.h[4][3]=b.h[4][4]=1;
	b.h[2][5]=b.h[5][5]=1;
	scanf("%lld",&n);
	power(n-1);
    printf("%lld",a.h[1][5]);	
}