codeforces Round 66 D. Array Splitting 数组分割k段(价值=∑a[i]*序列号) 求最大价值+后缀和

题目链接：http://codeforces.com/contest/1175/problem/D
题目大意：
把一个有n个元素的数组分割成k段，不能为空。价值=∑（每段的和*段号）。
求最大的价值。

例一：[-1 -2] [5 -4 8]
价值= -3 * 1 + 9 * 2 = 15

思路：开始想的贪心，在满足前面能够分的情况下，让最后一段尽可能大。以此类推。

k=1时特判，剩余的全部相加。最大值用rmq维护。

然后WA6.找到一个特例：-100 -100 2 1
只考虑了让加的最大，没有考虑让前面减的最少。

正确思路：ans=∑a[i]和，再把n到2的前缀和求出来，直接选取k-1个就可以了s=∑k个前缀和。

结果就是：ans+s。

证明：

WA：rmq+后缀和
#include<bits/stdc++.h>
#define LL long long
//#define p1 first
//#define p2 second
using namespace std;

LL s[300010]={0};

#define MAXN 300010
const LL N_N=log2(MAXN)+2;
LL n;
LL d[MAXN][N_N];
LL a[MAXN]={0};

void st()
{
    LL m, i, j;
    for(LL i=1; i<=n; i++)
        d[i][0]=a[i];

    m=log2(n);
    for (j = 1; j <= m; ++j)
    {
        for (i = 1; i + (1 << j) - 1 <= n; ++i)
            d[i][j] = max(d[i][j - 1], d[i + (1 << (j - 1))][j - 1]);
    }
}

LL rmq(LL a, LL b)
{
    LL m = (LL) (log(b - a + 1.0) / log(2.0));
    return max(d[a][m], d[b - (1 << m) + 1][m]);
}

int main()
{
    LL k;
    scanf("%lld%lld",&n,&k);
    for(LL i=1;i<=n;i++)
    {
        scanf("%lld",&s[n-i+1]);
    }
    for(LL i=1;i<=n;i++)
    {
        a[i]=a[i-1]+s[i];
    }
    st();
    LL R=1;
    LL ans=0, now_=0;
    while(k)
    {
        if(k==1)
        {
            for(LL i=R;i<=n;i++)
            {
                ans+=s[i];
            }
            k=0;
            break;
        }
        LL L=n-k+1;
        LL now=rmq(R, L)-now_;
        ans+=now*k;

        LL Ans=0;
        for(LL i=R;i<=L;i++)
        {
            Ans+=s[i];
            if(Ans==now)
            {
                R=i+1;
                break;
            }
        }
        now_+=now;
        k--;
    }
    cout<<ans<<endl;

    return 0;
}

AC:后缀和贪心
#include <bits/stdc++.h>
#define LL long long
using namespace std;

LL a[300010];
LL s[300010]={0};
int main()
{
    int n, k;
    LL ans=0;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
        ans+=a[i];
    }
    for(int i=n;i>=2;i--)
    {
        s[i]=s[i+1]+a[i];
    }
    sort(s+2, s+1+n);
    for(int i=n;i>=n-k+2;i--)
    {
        ans+=s[i];
    }
    printf("%lld\n",ans);

    return 0;
}