欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

【CF 185A】Plant

程序员文章站 2022-06-04 12:10:00
...

【CF 185A】Plant

【CF 185A】Plant

解析:

       矩阵快速幂。

       令【CF 185A】Plant表示【CF 185A】Plant年后的上三角形个数,【CF 185A】Plant【CF 185A】Plant年后的三角形个数于是得到递推式:

       【CF 185A】Plant

       于是可转化成矩阵乘法用矩阵快速幂解决。

 

代码:

#include <bits/stdc++.h>
using namespace std;

const int mod=1e9+7;
const int Max=3;
long long n;
int ans[Max]={0,1,1},a[Max][Max];

inline void mul(int ans[],int a[Max][Max])
{
	int c[Max];
	memset(c,0,sizeof(c));
	for(int j=1;j<=2;j++)
	  for(int k=1;k<=2;k++)
	    c[j]=(c[j]+1ll*ans[k]*a[k][j]%mod)%mod;
	memcpy(ans,c,sizeof(c));
}

inline void mulself(int a[Max][Max])
{
	int c[Max][Max];
	memset(c,0,sizeof(c));
	for(int i=1;i<=2;i++)
	  for(int j=1;j<=2;j++)
	    for(int k=1;k<=2;k++)
	      c[i][j]=(c[i][j]+1ll*a[i][k]*a[k][j]%mod)%mod;
	memcpy(a,c,sizeof(c));
}

inline void solve()
{
	while(n)
	{
	  if(n&1) mul(ans,a);
	  n>>=1;
	  mulself(a);
	}
}

int main()
{
	scanf("%I64d",&n);
	a[1][1]=2,a[2][1]=1,a[2][2]=4;
	solve();
	printf("%d",ans[1]);
	return 0;
}