bzoj4547: Hdu5171 小奇的集合（矩陣乘法）

題目傳送門
。

解法：
矩陣乘法啊。
每次選最大的兩個數相加。跟斐波那契差不多啊。

用一個位置表示sum就可以了呀。

坑點：
如果第二大的為負數，要用最大的數把這個負數先加到非負數先。然後才能正常做。

程式碼實現：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;const ll mod=10000007;
struct node {ll a[5][5];node() {memset(a,0,sizeof(a));}}per;
node jc(node a,node b) {
    node c;
    for(int i=1;i<=3;i++)for(int j=1;j<=3;j++)for(int k=1;k<=3;k++)
        c.a[i][j]=(c.a[i][j]+a.a[i][k]*b.a[k][j])%mod;
    return c;
}ll a[110000];
node pow_mod(node a,int b) {
    node ans=per;
    while(b!=0) {if(b%2==1)ans=jc(ans,a);a=jc(a,a);b/=2;}
    return ans;
}
int main() {
    int n,k;scanf("%d%d",&n,&k);ll sum=0;
    for(int i=1;i<=n;i++){scanf("%lld",&a[i]);sum=(sum+a[i])%mod;}
    if(k==0) {printf("%lld\n",sum);return 0;}
    sort(a+1,a+1+n);ll f=a[n],s=a[n-1];
    while(s<0) {
        s+=f;k--;sum=(sum+s)%mod;
        if(k==0) {printf("%lld\n",sum);return 0;}
    }
    for(int i=1;i<=3;i++)per.a[i][i]=1;
    node A;
                A.a[1][2]=1;A.a[1][3]=1;
    A.a[2][1]=1;A.a[2][2]=1;A.a[2][3]=1;
                            A.a[3][3]=1;
    A=pow_mod(A,k);printf("%lld\n",(sum+s*A.a[1][3]%mod+f*A.a[2][3]%mod)%mod);
    return 0;
}