题目大意: 有一个大小为n的可重集S,小奇每次操作可以加入一个数a+b(a,b均属于S),求k次操作后它可获得的S的和的最大值。(数据保证这个值为非负数)
我的收获:讨论大法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;
}