Java TreeSet使用的内存是否少于PriorityQueue?

时间:2018-06-26 14:40:55

标签: java priority-queue treeset

我正在这样做Question,首先使用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 ;
}

2 个答案:

答案 0 :(得分:1)

基本上与使用的内存量无关,只是使用PriorityQueue的程序永无止境,不断向PriorityQueue添加元素。

由于您的条件if(list.size() == 0 || list.get(list.size() - 1) != val),它不会结束-永远不会满足,因此检查list的大小(永不更改)的while循环始终为真,因此每个循环执行从队列中取出1个元素,然后向其中添加3个元素。

答案 1 :(得分:1)

这与数据结构无关。第一种情况的参数组合可能导致此语句永远不成立:

if(list.size() == 0 || list.get(list.size() - 1) != val)

这意味着循环永远不会终止,并且对象q会增长,直到耗尽内存。例如,尝试致电:

solve(1,1,2,5)

在这种情况下,两个数据结构之间的内存差异没有影响。需要注意的是与随机访问和下一个/上一个指针有关的一些差异,但这与此处无关。