Leetcode刷題時想用優先隊列解決,在使用時碰到了怎麼排序的問題。我首先給自定義的類Node81實作Comparator接口,重寫了接口中的 compare() 方法。
然後定義了一個優先隊列 PriorityQueue queue = new PriorityQueue<>();,代碼如下,然後運作時就報錯啦。
class Node81 implements Comparator<Node81>{
int row;
int count;
Node81(){}
Node81(int row, int count) {
this.row = row;
this.count = count;
}
@Override
public int compare(Node81 o1, Node81 o2) {
if (o1.count > o2.count) {
return 1;
}
if (o1.count < o2.count) {
return -1;
}
return o1.row > o2.row ? 1 : -1;
}
}
public int[] kWeakestRows(int[][] mat, int k) {
PriorityQueue<Node81> queue = new PriorityQueue<>();
int n = mat.length, m = mat[0].length;
for (int i = 0; i < mat.length; i++) {
int count = 0;
for (int j = 0; j < mat[i].length; j++) {
if (mat[i][j] == 1) {
++count;
}
}
Node81 node81 = new Node81(i, count);
queue.add(node81);
}
int[] ans = new int[k];
for (int i = 0; i < k; i++) {
Node81 remove = queue.remove();
ans[i] = remove.row;
}
return ans;
}
![](https://img.laitimes.com/img/9ZDMuAjOiMmIsIjOiQnIsIyZuBnLmVjZ3UWYygzY0UzNzczMhhzY5QzNjZWYxIzYxUmNhJ2Lc52YucWbp5GZzNmLn9Gbi1yZtl2Lc9CX6MHc0RHaiojIsJye.png)
經過調試後,發現這裡我犯了一個低級錯誤,我雖然實作了Comparator接口,但是在申明初始化優先隊列( PriorityQueue queue = new PriorityQueue<>();)的時候并沒有将比較器傳進去,是以在優先隊列添加節點時,因為 compartor 為空會進入 下圖中的 siftUpComparable() 方法,因為我沒有給Node81類實作 **Comparable **接口中的 compareTo() 方法,就會報上圖中的錯誤。
Comparable 接口
public interface Comparable<T> {
/**
* 傳回 1 :表示比 o 大
* 傳回 -1 :表示比 o 小
* 傳回 0 :表示和 o 相等
*/
public int compareTo(T o);
}
可以看到,該接口中其實就隻有一個 compareTo(T o) 的方法,将此對象與指定的對象進行順序比較。Comparable表示可被排序的,實作該接口的類的對象自動擁有排序功能,像Collections.sort()或Arrays.sort可以直接對遊該類組成的數組或者集合進行排序.
Comparator 接口
@FunctionalInterface
public interface Comparator<T> {
int compare(T o1, T o2);
boolean equals(Object obj);
// 1.8 後有很多新增的預設方法
}
Comparator是比較器,它可以作為一個參數傳遞到Collections.sort和Arrays.sort方法等來指定某個類對象的排序方式。像優先隊列就可以在初始化的時候給其傳遞一個實作了Camparator接口的類。如下方代碼所示。
// 隻要是實作了Comparator的接口都行
public PriorityQueue(Comparator<? super E> comparator) {
this(DEFAULT_INITIAL_CAPACITY, comparator);
}
是以我開始給優先隊列初始化應該給其加上參數:PriorityQueue queue = new PriorityQueue<>(new Node81()); 這樣優先隊列就會識别到有比較強而不去找compareTo() 方法。從這裡也可以看出:當Comparable和Comparator接口都實作了的時候,會優先使用Comparator接口的 compare() 方法來進行排序。
總結
- Comparable接口和Comparator的功能都差不多,都是給類排序
- Comparable是讓類能夠像基礎資料一樣自然排序,Comparator在使用時需要将其作為參數傳遞進去
- 當Comparable和Comparator接口同時都實作了的時候,會優先使用Comparator接口的排序規則