C - 【浅紫】特殊生成树 (Gym - 102072C )(矩阵快速幂)

将NN个点排列成一个圆形,中间放置一个点固定为根节点,问特殊生成树的种类数。

特殊生成树:除根节点以外,其他节点只能与自己左右节点相连,或与根节点相连。

p.s.若节点的左右节点为同一个节点,向左或向右连接视为不同的生成树。

由于种类数可能过大,对1,000,000,0071,000,000,007取模。

当N=3N=3时,所有生成树表示如下:

Input

多组数据,请输入要文件结束。

每组数据包含一个整数N(0<N<109)N(0<N<109)。

Output

对于每组数据,输出一个整数,表示特殊生成树的种类数,答案对1,000,000,0071,000,000,007取模。

Example

Input

2
3

Output

5
16

由于此题数据较大要用到矩阵快速幂

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod=1e9+7;
struct juzhen
{
    ll data[3][3];
};
juzhen mul(juzhen a,juzhen b)
{
    juzhen c;
    memset(c.data,0,sizeof(c.data));
    for(int k=0;k<3;k++)
    {
        for(int i=0;i<3;i++)
        {
            if(!a.data[i][k]) continue;
            for(int j=0;j<3;j++)
            {
                if(!b.data[k][j]) continue;
                c.data[i][j]=(c.data[i][j]+a.data[i][k]*b.data[k][j]%mod)%mod;
            }
        }
    }
    return c;
}
juzhen mat_pow(juzhen a,ll n)
{
    juzhen res;
    memset(res.data,0,sizeof(res.data));
    res.data[0][0]=res.data[1][1]=res.data[2][2]=1;
    while(n)
    {
        if(n&1) res=mul(res,a);
        a=mul(a,a);
        n>>=1;
    }
    return res;
}
int main()
{
    juzhen a;
    memset(a.data,0,sizeof(a.data));
    a.data[0][0]=3;
    a.data[0][2]=a.data[1][0]=a.data[2][2]=1;
    a.data[0][1]=-1;
    ll n;
    while(cin>>n)
    {
        if(n==1)
        {
            cout<<1<<endl;
            continue;
        }
        juzhen ans;
        ans=mat_pow(a,n-2);
        juzhen g;
        g.data[0][0]=5;
        g.data[1][0]=1;
        g.data[2][0]=2;
        juzhen res;
        res=mul(ans,g);
        cout<<(res.data[0][0]+mod)%mod<<endl;//就是这个地方
    }
    return 0;
}
//3 -1 1 f(n-1)
//1 0  0 f(n-2)
//0 0  1   2

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

ZZ --瑞 hopeACMer

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值