CodeForces - 1395D - Boboniu Chats with Du 贪心

CodeForces - 1395D - Boboniu Chats with Du 贪心

题意：如果$a_i>m$，并且当天可以说话，则接下来$d$天不能说话。其余所有情况都能说话，计算能说话的$\sum a_i$的最大值

思路：大于$m$的如果取，则肯定取当前最大的，并且是从最后一个位置$n$开始取，每隔$d$个取一个

如何去利用前缀和：暴力枚举取$x$个大于$m$的，那么会消耗$(x-1)(d+1)+1$天，剩余的天数如果能取$<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;
}