1. 程式人生 > >【兩次過】【Comparator】846. 多關鍵字排序

【兩次過】【Comparator】846. 多關鍵字排序

給定 n 個學生( 1 到 n 編號)以及他們的考試成績,這裡有兩個關鍵字,考試成績以及學生學號。根據第一關鍵字對陣列進行排序(降序),如果第一關鍵字相同則根據第二關鍵字進行排序(升序).

樣例

給出 [[2,50],[1,50],[3,100]]
返回 [[3,100],[1,50],[2,50]]


解題思路:

注意寫Comparator介面時,返回-1的條件即為期望排序結果,如本題中返回-1的條件即是升序

複雜易懂版本:

public class Solution {
    /**
     * @param array: the input array
     * @return: the sorted array
     */
    public int[][] multiSort(int[][] array) {
        // Write your code here
        Arrays.sort(array, new Comparator<int[]>(){
            public int compare(int[] a, int[] b){
                //根據考試成績降序
                if(a[1] > b[1])
                    return -1;
                else if(a[1] < b[1])
                    return 1;
                else{ //根據學號升序
                    if(a[0] < b[0])
                        return -1;
                    else if(a[0] > b[0])
                        return 1;
                    else    
                        return 0;
                }
            }
        });
        
        return array;
    }
}

簡略老鳥版本:

public class Solution {
    /**
     * @param array: the input array
     * @return: the sorted array
     */
    public int[][] multiSort(int[][] array) {
        // Write your code here
        Arrays.sort(array, new Comparator<int[]>(){
            public int compare(int[] l, int[] r) {
                if (l[1] != r[1]) {//根據考試成績降序
                    return r[1] - l[1];
                }
                //根據學號升序
                return l[0] - r[0];
            }
        });
        return array;
    }
}