题意:给出三个数:n,d,m,然后给出一个长度为n的数组,你需要做的是将这个数组重新定义一个顺序使得所有可以加在一起的数字的和最大(当数组里面的一个数大于m,接下来的d个数字不能加进来)
AC代码:
package 练习;
import java.io.*;
import java.math.BigInteger;
import java.util.*;
public class Main {
static Scanner sc=new Scanner(System.in);
static long pre[];
static int top1,top2,n,d,m;
public static void main(String[] args) {
show();
// for(int t=sc.nextInt();t>0;t--) {
// show();
// }
}
private static void show() {
int n=sc.nextInt();
long d=sc.nextLong();
long m=sc.nextLong();
List<Long>a=new ArrayList<>();
a.add((long)0);
List<Long>b=new ArrayList<>();
b.add((long)0);
pre=new long [n+1];
for(int i=1;i<=n;i++){
long x=sc.nextLong();
if( x<=m ) a.add(x);//不被禁言的
else b.add(x);//要被禁言的
}
Collections.sort(a);
Collections.sort(b);
List<Long> c=new ArrayList<>();
c.add((long)0);
for(int i=a.size()-1;i>0;i--) {
c.add(a.get(i));
}
a=c;
for(int i=1;i<=n;i++) {
if(i<a.size())pre[i]=pre[i-1]+a.get(i);
else pre[i]=pre[i-1];
}
long sumn=0,ans=pre[n];
for(int j=1;j<b.size();j++){//被禁言几次
sumn+=b.get(b.size()-j);
long yu=(j-1)*(d+1)+1;
if( yu>n ) break;
yu=n-yu;
ans=Math.max(ans,sumn+pre[(int)yu] );
}
System.out.println(ans);
}
}