-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathberries.txt
68 lines (56 loc) · 1.95 KB
/
berries.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
import java.io.*;
import java.util.*;
public class berries {
public static void main(String[] args) throws IOException {
BufferedReader f = new BufferedReader(new FileReader("berries.in"));
PrintWriter out = new PrintWriter(new BufferedWriter(new FileWriter("berries.out")),true);
StringTokenizer st = new StringTokenizer(f.readLine());
int treeNum = Integer.parseInt(st.nextToken());
int bucketNum=Integer.parseInt(st.nextToken());
st = new StringTokenizer(f.readLine());
int[] trees = new int[treeNum];
for(int i=0; i<treeNum; i++){
trees[i]=Integer.parseInt(st.nextToken());
}
int cum=-1;
for(int j: trees){
if(j>cum){
cum=j;
}
}
HashSet<Integer> answers = new HashSet<>();
for(int i=0; i<=cum;i++){
answers.add(verify(i,bucketNum,trees));
}
out.println(Collections.max(answers));
out.close();
}
public static int verify(int target, int bucketNum, int[] trees){
ArrayList<Integer> buckets = new ArrayList<>();
PriorityQueue<Integer> queue = new PriorityQueue<>(new Comparator<Integer>() {
@Override
public int compare(Integer integer, Integer t1) {
return -1*Integer.compare(integer,t1);
}
});
for(int i: trees){
queue.add(i);
}
while (buckets.size()<bucketNum){
int j=queue.poll();
if(j<target){
buckets.add(j);
queue.add(0);
}
else {
buckets.add(target);
queue.add(j-target);
}
}
int count =0;
for(int i=bucketNum-1; i>bucketNum/2-1; i--){
count+=buckets.get(i);
}
return count;
}
}