「省选模拟赛」[hdu5171] 小奇的集合
注:原题没有考虑最大值为负数的情况
「题目背景」
小奇总是在数学课上思考奇怪的问题。
「问题描述」
有一个大小为n的可重集S,小奇每次操作可以加入一个数a+b(a,b均属于S),求k次操作后它可获得的S的和的最大值。(数据保证这个值为非负数)
「输入格式」
第一行有两个整数n,k表示初始元素数量和操作数,第二行包含n个整数表示初始时可重集的元素。
「输出格式」
输出一个整数,表示和的最大值。答案对10000007取模。
「样例输入」
2 2
3 6
「样例输出」
33
「数据范围」
对于30%的数据,有 n<=10^5,k<=10^5,|ai|<=10^5。
对于100%的数据,有 n<=10^5,k<=10^9,|ai|<=10^5。
题解
每次显然是选择最大和次大的元素相加,如果相加得到的数是集合的最大值,很容易就能写出递推式
K<=10^9显然是矩阵乘法加速
0 1 0 Fi-1 Fi
1 1 0 Fi => Fi+1
1 1 1 S S’
特殊情况是集合的最大值和次大值可能为负数
1.只有次大值为负数,模拟若干次至其为正数再递推
2.两个都为负数,直接计算(每次都是这两个相加)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
#include<map> #include<set> #include<cmath> #include<cstdio> #include<cstring> #include<cstdlib> #include<iostream> #include<algorithm> #define inf 1000000000 #define ll long long #define mod 10000007 using namespace std; ll read() { ll x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } int n,K; int val[300005]; ll tot,mn,mx; struct M{ ll v[5][5]; M(){ memset(v,0,sizeof(v)); } friend M operator*(M a,M b){ M c; for(int i=1;i<=3;i++) for(int j=1;j<=3;j++) for(int k=1;k<=3;k++) c.v[i][j]=(c.v[i][j]+a.v[i][k]*b.v[k][j])%mod; return c; } friend M operator^(M a,int b){ M ans; for(int i=1;i<=3;i++) ans.v[i][i]=1; for(int i=b;i;i>>=1,a=a*a) if(i&1)ans=ans*a; return ans; } }; int main() { freopen("set.in","r",stdin); freopen("set.out","w",stdout); n=read();K=read(); for(int i=1;i<=n;i++) val[i]=read(); for(int i=1;i<=n;i++) tot+=val[i],tot%=mod; sort(val+1,val+n+1); if(val[n]<=0) { tot+=((ll)(val[n]+val[n-1]))*K; printf("%d\n",tot%mod); } else { while(val[n-1]<0&&K) { val[n-1]=(val[n]+val[n-1])%mod; tot=(tot+val[n-1])%mod; K--; } M a,b; b.v[1][2]=b.v[2][1]=b.v[2][2]=b.v[3][1]=b.v[3][2]=b.v[3][3]=1; a.v[1][1]=val[n-1];a.v[2][1]=val[n];a.v[3][1]=tot; a=(b^K)*a; int ans=a.v[3][1]; if(ans<0)ans+=mod; printf("%d\n",ans); } return 0; } |
Subscribe