web-dev-qa-db-ja.com

priorityQueueをmax priorityqueueに変更します

整数のJavaに優先キューがあります：

 PriorityQueue<Integer> pq= new PriorityQueue<Integer>();

pq.poll()を呼び出すと、最小要素が取得されます。

質問：コードを変更して最大の要素を取得する方法は？

javacollectionspriority-queue
90
Borut Flis

これはどうですか：

PriorityQueue<Integer> queue = new PriorityQueue<>(10, Collections.reverseOrder());
queue.offer(1);
queue.offer(2);
queue.offer(3);
//...

Integer val = null;
while( (val = queue.poll()) != null) {
    System.out.println(val);
}

Collections.reverseOrder()はComparatorを提供し、この場合、PriorityQueueの要素を正順で自然順序にソートします。

178
Edwin Dalorzo

Java 8以降のラムダ式を使用できます。

次のコードでは、10が大きくなります。

PriorityQueue<Integer> pq = new PriorityQueue<>((x, y) -> y - x);
pq.add(10);
pq.add(5);
System.out.println(pq.peek());

ラムダ関数は、入力パラメーターとして2つの整数を受け取り、それらを互いに減算し、算術結果を返します。ラムダ関数は、機能インターフェイスComparator<T>を実装します。 （これは、匿名クラスまたは個別の実装とは対照的に、所定の場所で使用されます。）

50
Guangtong Shen

要素を逆順でランク付けするカスタムComparatorオブジェクトを提供できます。

PriorityQueue<Integer> pq = new PriorityQueue<Integer>(defaultSize, new Comparator<Integer>() {
    public int compare(Integer lhs, Integer rhs) {
        if (lhs < rhs) return +1;
        if (lhs.equals(rhs)) return 0;
        return -1;
    }
});

これで、優先度キューはすべての比較を逆にするため、最小要素ではなく最大要素が取得されます。

お役に立てれば！

21
templatetypedef
PriorityQueue<Integer> pq = new PriorityQueue<Integer> (
  new Comparator<Integer> () {
    public int compare(Integer a, Integer b) {
       return b - a;
    }
  }
);
12
Varun Juneja

優先度キューの要素は、自然な順序に従って、またはキューの構築時に提供されるコンパレータによって順序付けられます。

コンパレータは比較メソッドをオーバーライドする必要があります。

int compare(T o1, T o2)

デフォルトの比較メソッドは、最初の引数が2番目の引数より小さい、等しい、または大きいため、負の整数、ゼロ、または正の整数を返します。

Javaが提供するデフォルトのPriorityQueueはMin-Heapです。最大ヒープが必要な場合は、次のコードを使用します。

public class Sample {
    public static void main(String[] args) {
        PriorityQueue<Integer> q = new PriorityQueue<Integer>(new Comparator<Integer>() {

            public int compare(Integer lhs, Integer rhs) {
                if(lhs<rhs) return +1;
                if(lhs>rhs) return -1;
                return 0;
            }
        });
        q.add(13);
        q.add(4);q.add(14);q.add(-4);q.add(1);
        while (!q.isEmpty()) {
            System.out.println(q.poll());
        }
    }

}

リファレンス： https://docs.Oracle.com/javase/7/docs/api/Java/util/PriorityQueue.html#comparator（）

8
rohan

JavaのMax-Heapのサンプルは次のとおりです。

PriorityQueue<Integer> pq1= new PriorityQueue<Integer>(10, new Comparator<Integer>() {
public int compare(Integer x, Integer y) {
if (x < y) return 1;
if (x > y) return -1;
return 0;
}
});
pq1.add(5);
pq1.add(10);
pq1.add(-1);
System.out.println("Peek: "+pq1.peek());

出力は10になります

4
Nobal

MinMaxPriorityQueue（Guavaライブラリの一部）を使用できます： ドキュメントはこちら 。 poll()の代わりに、pollLast()メソッドを呼び出す必要があります。

3
Chthonic Project

次のようなものを試すことができます：

PriorityQueue<Integer> pq = new PriorityQueue<>((x, y) -> -1 * Integer.compare(x, y));

あなたが持っているかもしれない他のベース比較関数のために機能します。

1
Horia Coman

ダブルヒープソートmin maxで両方のコンパレータでモンテカルロシミュレーションを実行したところ、両方とも同じ結果になりました。

これらは私が使用した最大のコンパレータです：

（A）コレクションの組み込みコンパレータ

 PriorityQueue<Integer> heapLow = new PriorityQueue<Integer>(Collections.reverseOrder());

（B）カスタムコンパレータ

PriorityQueue<Integer> heapLow = new PriorityQueue<Integer>(new Comparator<Integer>() {
    int compare(Integer lhs, Integer rhs) {
        if (rhs > lhs) return +1;
        if (rhs < lhs) return -1;
        return 0;
    }
});
1
sivi

PriorityQueueをMAX PriorityQueueに変更する方法1：キューpq = new PriorityQueue <>（Collections.reverseOrder（））;方法2：キューpq1 =新しいPriorityQueue <>（（a、b）-> b-a）;いくつかの例を見てみましょう。

public class Example1 {
    public static void main(String[] args) {

        List<Integer> ints = Arrays.asList(222, 555, 666, 333, 111, 888, 777, 444);
        Queue<Integer> pq = new PriorityQueue<>(Collections.reverseOrder());
        pq.addAll(ints);
        System.out.println("Priority Queue => " + pq);
        System.out.println("Max element in the list => " + pq.peek());
        System.out.println("......................");
        // another way
        Queue<Integer> pq1 = new PriorityQueue<>((a, b) -> b - a);
        pq1.addAll(ints);
        System.out.println("Priority Queue => " + pq1);
        System.out.println("Max element in the list => " + pq1.peek());
        /* OUTPUT
          Priority Queue => [888, 444, 777, 333, 111, 555, 666, 222]
          Max element in the list => 888
          ......................
           Priority Queue => [888, 444, 777, 333, 111, 555, 666, 222]
           Max element in the list => 888

         */


    }
}

有名なインタビューの問題を取り上げましょう：PriorityQueueを使用した配列のK番目の最大要素

public class KthLargestElement_1{
    public static void main(String[] args) {

        List<Integer> ints = Arrays.asList(222, 555, 666, 333, 111, 888, 777, 444);
        int k = 3;
        Queue<Integer> pq = new PriorityQueue<>(Collections.reverseOrder());
        pq.addAll(ints);
        System.out.println("Priority Queue => " + pq);
        System.out.println("Max element in the list => " + pq.peek());
        while (--k > 0) {
            pq.poll();
        } // while
        System.out.println("Third largest => " + pq.peek());
/*
 Priority Queue => [888, 444, 777, 333, 111, 555, 666, 222]
Max element in the list => 888
Third largest => 666

 */
    }
}

別の方法 ：

public class KthLargestElement_2 {
    public static void main(String[] args) {
        List<Integer> ints = Arrays.asList(222, 555, 666, 333, 111, 888, 777, 444);
        int k = 3;

        Queue<Integer> pq1 = new PriorityQueue<>((a, b) -> b - a);
        pq1.addAll(ints);
        System.out.println("Priority Queue => " + pq1);
        System.out.println("Max element in the list => " + pq1.peek());
        while (--k > 0) {
            pq1.poll();
        } // while
        System.out.println("Third largest => " + pq1.peek());
        /*
          Priority Queue => [888, 444, 777, 333, 111, 555, 666, 222] 
          Max element in the list => 888 
          Third largest => 666

         */
    }
}

ご覧のとおり、どちらも同じ結果をもたらしています。

1
Soudipta Dutta

これは、コンパレーターのみを使用するコンストラクターを導入したJava 8の以下のコードによって実現できます。

PriorityQueue<Integer> maxPriorityQ = new PriorityQueue<Integer>(Collections.reverseOrder());
0
Anshu Shekhar
PriorityQueue<Integer> lowers = new PriorityQueue<>((o1, o2) -> -1 * o1.compareTo(o2));
0
Sumit Kumar Saha

逆符号を使用して要素をプッシュしてみてください。例：a = 2＆b = 5を追加してから、b = 5をポーリングします。

PriorityQueue<Integer>  pq = new PriorityQueue<>();
pq.add(-a);
pq.add(-b);
System.out.print(-pq.poll());

キューの先頭をポーリングしたら、使用法の記号を逆にします。これにより、5（より大きな要素）が印刷されます。単純な実装で使用できます。確かに信頼できる修正ではありません。お勧めしません。

0
striker28