题目描述
给你
n
个数,让你取出其中的某些数,使得剩下的数组成一个序列(不能调换顺序),序列的
解题思路
多次贪心取最优值。
假设当前选的序列的前缀和为
sumi
。
1.如果一个数
aj
在上一个序列没有选,但是
aj∗numj+sumn−sumj≥0
,
numj
表示
aj
排在序列的第几位。那么这样的话选这个数一定更优,因为如果选了
aj
,那么从
j+1
到
n
选的数的权值都会增加,总共增加
2.如果一个数 aj 在上一个序列有选,但是 aj∗numj+sumn−sumj<0 ,一定不选更优。
参考代码
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
#define maxn 100005
#define ll long long
using namespace std;
ll a[maxn],s[maxn],ans,n;
bool cho[maxn];
int read(){
int ret=0,ff=1;
char ch=getchar();
while (ch<'0' || ch>'9') {
if (ch=='-') ff=-1;
ch=getchar();
}
while (ch>='0' && ch<='9') {
ret=ret*10+ch-'0';
ch=getchar();
}
return ret*ff;
}
int main(){
n=read();
fo(i,1,n) {
a[i]=read();
cho[i]=1;
s[i]=s[i-1]+a[i];
}
for(;;){
int tot=1;
bool bz=0;
ll now=0;
fo(i,1,n) {
if (cho[i] && a[i]*tot+s[n]-s[i]<0) {
cho[i]=0;
bz=1;
}
else if (!cho[i] && a[i]*tot+s[n]-s[i]>=0) {
cho[i]=1;
bz=1;
}
if (cho[i]) {
now+=a[i]*tot;
tot++;
s[i]=s[i-1]+a[i];
}
else s[i]=s[i-1];
}
ans=now;
if (!bz) break;
}
cout<<ans;
return 0;
}