CodeForces - 1395D - Boboniu Chats with Du 贪心
题意:如果 a i > m a_i>m ai>m,并且当天可以说话,则接下来 d d d天不能说话。其余所有情况都能说话,计算能说话的 ∑ a i \sum a_i ∑ai的最大值
思路:大于 m m m的如果取,则肯定取当前最大的,并且是从最后一个位置 n n n开始取,每隔 d d d个取一个
如何去利用前缀和:暴力枚举取 x x x个大于 m m m的,那么会消耗 ( x − 1 ) ( d + 1 ) + 1 (x-1)(d+1)+1 (x−1)(d+1)+1天,剩余的天数如果能取 < m <m <m的,全部取 < m <m <m的肯定是最优的
代码:
int n,d,m,x,a[maxn],b[maxn],m1=0,m2=0;
ll sum1[maxn],sum2[maxn];
bool cmp(int a,int b){return a>b;}
int main()
{
scanf("%d%d%d",&n,&d,&m);
rep(i,1,n)
{
scanf("%d",&x);
if (x>m)a[++m1]=x;
else b[++m2]=x;
}
sort(a+1,a+1+m1,cmp);
sort(b+1,b+1+m2,cmp);
rep(i,1,m1)sum1[i]=sum1[i-1]+a[i];
rep(i,1,m2)sum2[i]=sum2[i-1]+b[i];
ll ans=sum2[m2];
rep(i,1,m1)
{
int day=(i-1)*(d+1)+1;
if (day>n)break;
ans=max(ans,sum1[i]+sum2[min(n-day,m2)]);
}
WW(ans);
return 0;
}