4547: Hdu5171 小奇的集合

题目链接

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

题解：mod GXZlegend

我的收获：讨论大法2333

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

#define rep(x,y) for(int x=0;x<y;x++)
#define N 3
#define ll long long
#define P 10000007

ll n,k,a[100005];
ll sum;

struct Matrix{
    int x,y;ll v[N][N];
    ll *operator [] (int x){return v[x];}
    Matrix(int _x,int _y,int w=0){
        x=_x,y=_y;
        for(int i=0;i<3;i++) for(int j=0;j<3;j++) v[i][j]=(i==j)?w:0;
    }
};

Matrix operator * (Matrix a,Matrix b){
    Matrix c(a.x,b.y);
    for(int i=0;i<3;i++) for(int j=0;j<3;j++) for(int k=0;k<3;k++) c[i][j]=(c[i][j]+a[i][k]*b[k][j])%P;
    return c;
}

Matrix operator ^ (Matrix a,int b){
    Matrix c(a.x,a.y,1);
    for(;b;b>>=1,a=a*a)
    if(b&1) c=c*a;
    return c;
}

void work()
{
    if(a[n]<=0) printf("%lld\n",((sum+(a[n]+a[n-1])*k)%P+P)%P);
    else
    {
        for(sum-=a[n];k&&a[n-1]<0;k--) a[n-1]+=a[n],sum+=a[n-1];
        Matrix A(3,3);A[0][0]=A[0][1]=A[0][2]=A[1][0]=A[2][2]=1,A=A^(k+1);
        printf("%lld\n",(((sum+a[n]*A[0][2]+a[n-1]*A[1][2])%P+P)%P));
    }
}

void init()
{
    scanf("%lld%lld",&n,&k);
    for(int i=1;i<=n;i++) scanf("%lld",&a[i]),sum+=a[i];
    sort(a+1,a+n+1);
}

int main()
{
    init();
    work();
    return 0;
}