我正在做这个问题并首先使用PriorityQueue解决了它:-
public ArrayList<Integer> solve(int A, int B, int C, int D) {
PriorityQueue<Integer> q = new PriorityQueue<>();
q.add(A);
q.add(B);
q.add(C);
ArrayList<Integer> list = new ArrayList<>();
while(list.size() < D){
int val = q.poll();
if(list.size() == 0 || list.get(list.size() - 1) != val)
list.add(val);
q.add(val*A);
q.add(val*B);
q.add(val*C);
}
list.sort(null);
return list;
}
但是它给出了java.lang.OutOfMemoryError:Java堆空间错误。
用TreeSet替换PriorityQueue后,该解决方案被接受:-
public ArrayList<Integer> solve(int A, int B, int C, int D) {
ArrayList<Integer> res = new ArrayList<>() ;
TreeSet<Integer> set = new TreeSet<>() ;
set.add(A) ;
set.add(B) ;
set.add(C) ;
for(int i = 0; i < D; i++) {
int temp = set.first() ;
set.remove(temp) ;
res.add(temp) ;
set.add(temp*A) ;
set.add(temp*B) ;
set.add(temp*C) ;
}
return res ;
}
jeck猫
收到一只叮咚
相关分类