Plant(矩阵快速幂(推理))

【题目来源】https://vjudge.net/problem/CodeForces-185A
【题意】
每一个小三角形每次都会变成四个小小三角形,分别有3个朝上的,1个朝下的,问,第n次的图形里有多少个朝上的三角形。
【思路】
第一开始忽略了图形上的规律,妄图直接找数字上的关系,结果想当然的失败了。
然后呢,换了方向,发现,一个三角形总会分成四个小小三角形,具体来说,一个朝上的三角形会分成三个朝上的和一个朝下的,一个朝下的会分成一个朝上的和三个朝上的。
假设当前三角形拥有a个朝上的三角形和b个朝下的三角形,那么下一个图形会变成拥有3a+b个朝上的三角形, 和a+3b个朝下的三角形。
由此可以推出中间矩阵。
使用矩阵关系表达出来即是:
这里写图片描述
【代码】

#include<set>
#include<map>
#include<stack>
#include<cmath>
#include<queue>
#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<limits.h>
#include<algorithm>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
const int mod=1e9+7;
const double esp=1e-5;
typedef unsigned long long ll;
typedef long long LL;
LL k;
struct mat
{
    LL a[3][3];
};
mat operator*(mat s,mat t)
{
    mat r;
    mem(r.a,0);
    for(int i=1; i<=2; i++)
    {
        for(int j=1; j<=2; j++)
        {
            for(int p=1; p<=2; p++)
            {
                r.a[i][j]=r.a[i][j]+s.a[i][p]*t.a[p][j];
                if(r.a[i][j]>=mod)
                    r.a[i][j]%=mod;
            }
        }
    }
    return r;
}
//void print(mat s)
//{
//    for(int i=1;i<=2;i++)
//    {
//        for(int j=1;j<=2;j++)
//            {
//                printf("%d ",s.a[i][j]);
//            }
//            printf("\n");
//    }
//}
void pow_mat(mat &ans)
{
    mat base,temp;
    mem(base.a,0);
    base.a[1][1]=3;
    base.a[1][2]=1;
    base.a[2][1]=1;
    base.a[2][2]=3;
    mem(temp.a,0);
    for(int i=1; i<=2; i++)
        temp.a[i][i]=1;
    while(k)
    {
        if(k&1)
            temp=temp*base;
        base=base*base;
        k>>=1;
    }
//    print(temp);
    ans=ans*temp;
}
int main()
{
    while(cin>>k)
    {
        if(k==0)
        {
            cout<<"1"<<endl;
            continue;
        }
        mat ans;
        mem(ans.a,0);
        ans.a[1][1]=1;
        ans.a[1][2]=0;
        pow_mat(ans);
        cout<<ans.a[1][1]<<endl;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值