CodeForces - 1395D - Boboniu Chats with Du 贪心

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 (x1)(d+1)+1天,剩余的天数如果能取 < 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;
}

你可能感兴趣的:(Codeforces)