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