题目链接: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;
}