【矩阵】HDU 5171 GTY's birthday gift 快速幂

点击打开链接

取最大的两个 构造一个矩阵， 这里构造一个3*3 的 （比较明显

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <string>
#include <iostream>
#include <algorithm>
#include <sstream>
#include <cmath>
using namespace std;
#include <queue>
#include <stack>
#include <set>
#include <vector>
#include <deque>
#include <map>
#define cler(arr, val)    memset(arr, val, sizeof(arr))
typedef long long  LL;
const int MAXN = 714100;
const int MAXM = 140000;
const int INF = 0x3f3f3f3f;
const LL mod = 10000007;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
LL a,n,k,d[2],sum=0;
struct mat
{
    LL s[4][4];
};
void operator*=( mat &a,const mat &b)
        {
            static mat c;
            int i,j,k;
            memset(c.s,0,sizeof(c));
            for(k=0;k<3;k++)
                for(i=0;i<3;i++)
                    for(j=0;j<3;j++)
                        c.s[i][j]+=a.s[i][k]*b.s[k][j]%mod,c.s[i][j]%=mod;
            for(i=0;i<3;i++)
                for(j=0;j<3;j++)
                    a.s[i][j]=c.s[i][j];
        }
mat ans,t;
int main()
{
    LL n,k;
    while(cin>>n>>k)
    {
        cler(d,0);
        sum=0;
        for(int i=1;i<=n;i++)
        {
            cin>>a;
            if(a>=d[0])
            {
                swap(d[0],d[1]);
                d[0]=a;
            }
            else if(a>d[1])
                d[1]=a;
            sum+=a;
        }
        t.s[0][0]=1;
        t.s[0][1]=0;
        t.s[0][2]=0;

        t.s[1][0]=1;
        t.s[1][1]=1;
        t.s[1][2]=1;

        t.s[2][0]=1;
        t.s[2][1]=1;
        t.s[2][2]=0;

        cler(ans.s,0);
        ans.s[0][0]=sum;
        ans.s[0][1]=d[0];
        ans.s[0][2]=d[1];
        for(LL i=k;i;i=i>>1,t*=t)
        {
            if(i&1)
                ans*=t;
        }
        printf("%I64d\n",ans.s[0][0]);
//        cout<<ans.s[0][0]%mod<<endl;
    }
    return 0;
}