【兩次過】【Comparator】846. 多關鍵字排序
阿新 • • 發佈:2018-12-24
給定 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; } }