[bzoj 4547]小奇的集合

有一个大小为n的可重集S，小奇每次操作可以加入一个数a+b(a,b均属于S)，求k次操作后它可获得的S的和的最大值。（数据保证这个值为非负数）

这道题一看到它的范围这么大，操作的方法又那么单一，于是便想到了矩阵乘法。相乘的方式为{1,1,1}{0,0,1}{0,1,1}*{s}{m1}[m2}={s+m1+m2}{m2}{m1+m2}（s为当前的总和，m1为次大值，m2为最大值）。但有一个坑点，如果次大值为负数，那么相加后的值就不是新的最大值，就不满足矩阵相乘的含义，所以一开始要先暴力一下。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#define mod 10000007
using namespace std;
struct node
{
    long long a[5][5];
    node(){memset(a,0,sizeof(a));}
};
int s[100010];
node chengfa1(node a,node b)
{
    node c;
    for(int i=1;i<=3;i++)
    {
        for(int k=1;k<=3;k++)
        {
            c.a[i][1]=(c.a[i][1]+a.a[i][k]*b.a[k][1])%mod;
        }
    }
    return c;
}
node chengfa2(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;
}
int main()
{
    int n,k;long long sum=0,max1=0,max2=0;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&s[i]);
        sum+=s[i];
    }
    sort(s+1,s+n+1);
    max1=s[n],max2=s[n-1];
    while(1)
    {
        if(max1>=0 && max2>=0)break;
        int wy=max1+max2;
        if(wy>max1)max1=wy;
        else max2=wy;
        sum+=wy;k--;
        if(k==0){printf("%lld\n",sum);return 0;}
    }
    node pre,ss;
    pre.a[1][1]=pre.a[1][2]=pre.a[1][3]=pre.a[2][3]=pre.a[3][2]=pre.a[3][3]=1;
    ss.a[1][1]=sum;ss.a[2][1]=max2;ss.a[3][1]=max1;
    while(k)
    {
        if(k%2==1)ss=chengfa1(pre,ss);
        pre=chengfa2(pre,pre);
        k/=2;
    }
    printf("%lld\n",ss.a[1][1]%mod);
    return 0;
}