hdu5171 GTY's birthday gift 矩阵快速幂求斐波那契前n项和,矩阵快速幂模板

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<bitset>
#define fi first
#define se second
#define pii pair<int,int>
#define ll long long
#define inf 1<<30
#define eps 1e-8
using namespace std;
const int maxn=100005;
const ll mod=10000007;
int n;
ll k;
int d[maxn];
struct mat
{
    ll e[3][3];
};
mat mul(mat p,mat q)
{
    mat c;
    for(int i=0;i<3;i++)
    {
        for(int j=0;j<3;j++)
        {
            c.e[i][j]=0;
            for(int k=0;k<3;k++)
            {
                c.e[i][j]=(c.e[i][j]+p.e[i][k]*q.e[k][j]%mod)%mod;
            }
        }
    }
    return c;
}
mat powMat(mat u,ll b)
{
    mat s;
    memset(s.e,0,sizeof(s));
    s.e[0][0]=s.e[1][1]=s.e[2][2]=1;
    while(b)
    {
        if(b&1)
            s=mul(s,u);
        u=mul(u,u);
        b=(b>>1);
    }
    return s;
}
int main()
{
    while(~scanf("%d%I64d",&n,&k)) {
        ll sum=0;
        for(int i=0;i<n;i++) {
            scanf("%d",&d[i]);
            sum=(sum+d[i])%mod;
        }
        sort(d,d+n);
        mat m;
        m.e[0][0]=1; m.e[0][1]=1; m.e[0][2]=1;
        m.e[1][0]=0; m.e[1][1]=1; m.e[1][2]=1;
        m.e[2][0]=0; m.e[2][1]=1; m.e[2][2]=0;
        mat u=powMat(m,k);
        sum=(sum+u.e[0][1]*d[n-1]%mod+u.e[0][2]*d[n-2]%mod)%mod;
        printf("%I64d\n",sum);
    }
    return 0;
}

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值