codeforces 573E

题目描述

给你n 个数，让你取出其中的某些数，使得剩下的数组成一个序列(不能调换顺序)，序列的∑ki=1i∗si 最大。

解题思路

多次贪心取最优值。

假设当前选的序列的前缀和为sumi 。
1.如果一个数aj 在上一个序列没有选，但是aj∗numj+sumn−sumj≥0 ,numj 表示aj 排在序列的第几位。那么这样的话选这个数一定更优，因为如果选了aj ,那么从j+1 到n 选的数的权值都会增加，总共增加sumn−sumj ,如果这个增加量比选aj 的失去量要大，那么，选，一定更优。

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;
}