Сохранение набора деревьев Сохранение - Java

Question

У меня есть очередь приоритетов, которая содержит пару точек Я выполнял итерацию в этой очереди и вставлял каждую из точек в паре в набор деревьев.

Проблема в том, что я хочу поддерживать порядок, который был в очереди приоритетов Как мне реализовать Компаратор набора деревьев?

Answer 1

Вот возможность. Я не вставляю сами точки в TreeSet, но обертка, которая отслеживает порядок, в котором они находились, в PriorityQueue. Это имеет то преимущество, что нам не нужно изменять класс Point.

Вот класс-оболочка:

import java.util.PriorityQueue; 
import java.util.TreeSet; 

public class Sortable<T> implements Comparable<Sortable<T>> { 

    T element; 
    int order; 

    public Sortable(T element, int order) { 
     this.element = element; 
     this.order = order; 
    } 

    @Override 
    public int compareTo(Sortable<T> o) { 
     return order - o.order; 
    } 

} 

Теперь вы можете сделать:

public void transfer(PriorityQueue<PointPair> from, TreeSet<Sortable<Point>> to) { 
    PointPair pp = from.poll(); 
    int count = 0; 
    while (pp != null) { 
     to.add(new Sortable<Point>(pp.getLeftPoint(), count)); 
     count++; 
     to.add(new Sortable<Point>(pp.getRightPoint(), count)); 
     count++; 

     pp = from.poll(); 
    } 
}