天天看點

Java中的Comparable 和 Comparator接口

        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;
    }

           
Java中的Comparable 和 Comparator接口

     經過調試後,發現這裡我犯了一個低級錯誤,我雖然實作了Comparator接口,但是在申明初始化優先隊列( PriorityQueue queue = new PriorityQueue<>();)的時候并沒有将比較器傳進去,是以在優先隊列添加節點時,因為 compartor 為空會進入 下圖中的 siftUpComparable() 方法,因為我沒有給Node81類實作 **Comparable **接口中的 compareTo() 方法,就會報上圖中的錯誤。

Java中的Comparable 和 Comparator接口

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接口的排序規則