AC日记——3的幂的和 51nod 1013

3的幂的和

思路；

　　矩阵快速幂；

　   sn-1      3 1        sn

　　　   *          =

　　1　     0 1　　　  1

来，上代码：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

#define mod 1000000007

struct NodeType {
    long long a[][];
};
struct NodeType ans;

long long n;

struct NodeType mul(NodeType pos)
{
    NodeType res;
    res.a[][]=,res.a[][]=;
    res.a[][]=,res.a[][]=;
    for(int i=;i<=;i++)
    {
        for(int j=;j<=;j++)
        {
            for(int v=;v<=;v++)
            {
                res.a[i][j]=(res.a[i][j]+(pos.a[i][v]*pos.a[v][j])%mod)%mod;
            }
        }
    }
    return res;
}

void poww(int mi)
{
    NodeType pos;
    pos.a[][]=,pos.a[][]=;
    pos.a[][]=,pos.a[][]=;
    while(mi>)
    {
        struct NodeType res;
        res.a[][]=,res.a[][]=;
        res.a[][]=,res.a[][]=;
        if(mi&)
        {
            for(int i=;i<=;i++)
            {
                for(int j=;j<=;j++)
                {
                    for(int v=;v<=;v++)
                    {
                        res.a[i][j]=(res.a[i][j]+(ans.a[i][v]*pos.a[v][j])%mod)%mod;
                    }
                }
            }
            ans=res;
        }
        pos=mul(pos),mi=mi>>;
    }
}

int main()
{
    scanf("%lld",&n);n--;
    ans.a[][]=,ans.a[][]=;
    ans.a[][]=,ans.a[][]=;
    poww(n);
    cout<<(ans.a[][]+ans.a[][])%mod;
    return ;
}