【矩阵】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;
}