新らしくなった基本情報 科目 B 20問バージョン アルゴリズムとプログラミング サンプル問題8をJavaにしてみました

import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;

public class Q8 {
    //本問で問われている手続
    static void prioSched() {

        PriorityQueue<String> prioQueue = new PriorityQueue<>();

        prioQueue.enqueue("A", 1);
        prioQueue.enqueue("B", 2);
        prioQueue.enqueue("C", 2);
        prioQueue.enqueue("D", 3);
        prioQueue.dequeue();
        prioQueue.dequeue();
        prioQueue.enqueue("D", 3);
        prioQueue.enqueue("B", 2);
        prioQueue.dequeue();
        prioQueue.dequeue();
        prioQueue.enqueue("C", 2);
        prioQueue.enqueue("A", 1);

        while (prioQueue.size() != 0) {
            System.out.print(prioQueue.dequeue() + " ");
        }
    }

    public static void main(String[] args) {
        prioSched();
    }
}

class PriorityQueue<T> {

    private List<Data<T, Integer>> queue;

    public PriorityQueue() {
        queue = new ArrayList<>();
    }

    public void enqueue(T item, int priority) {
        Data<T, Integer> pair = new Data<>(item, priority);
        queue.add(pair);
        queue.sort(new PairComparator<>());
    }

    public T dequeue() {
        Data<T, Integer> pair = queue.remove(0);
        return pair.getString();
    }

    public int size() {
        return queue.size();
    }
}

class PairComparator<T> implements Comparator<Data<T, Integer>> {
    @Override
    public int compare(Data<T, Integer> p1, Data<T, Integer> p2) {
        return Integer.compare(p1.getPrio(), p2.getPrio());
    }
}

class Data<I, P> {
    private final I item;
    private final P prio;

    public Data(I string, P prio) {
        this.item = string;
        this.prio = prio;
    }

    public I getString() {
        return item;
    }

    public P getPrio() {
        return prio;
    }
}