无尽算法之 重新排列日志文件 (PriorityQueue)

你有一个日志数组 logs。每条日志都是以空格分隔的字串。

对于每条日志，其第一个字为字母数字标识符。然后，要么：

标识符后面的每个字将仅由小写字母组成，或；
标识符后面的每个字将仅由数字组成。
我们将这两种日志分别称为字母日志和数字日志。保证每个日志在其标识符后面至少有一个字。

将日志重新排序，使得所有字母日志都排在数字日志之前。字母日志按内容字母顺序排序，忽略标识符；在内容相同时，按标识符排序。数字日志应该按原来的顺序排列。

返回日志的最终顺序。

示例 ：

输入：[“a1 9 2 3 1”,“g1 act car”,“zo4 4 7”,“ab1 off key dog”,“a8 act zoo”]
输出：[“g1 act car”,“a8 act zoo”,“ab1 off key dog”,“a1 9 2 3 1”,“zo4 4 7”]

提示：

0 <= logs.length <= 100
3 <= logs[i].length <= 100
logs[i] 保证有一个标识符，并且标识符后面有一个字。

思路:

利用 PriorityQueue, 自定义 Comparator

题解:

class Solution {
    class MyComparator implements Comparator<String>{
    @Override
    public int compare(String s1, String s2) {
        int count = 1;
        String[] arr1 = s1.split(" ");
        String[] arr2 = s2.split(" ");
        while(arr1.length>count && arr2.length>count){
            if(arr1[count].compareTo(arr2[count])!=0){
                return arr1[count].compareTo(arr2[count]);
            }
            count++;
        }
        return arr1[0].compareTo(arr2[0]);
    }
}   
    public String[] reorderLogFiles(String[] logs) {
        PriorityQueue<String> p = new PriorityQueue<>(new MyComparator());
        ArrayList<String> res = new ArrayList<>();
        for(String s : logs){
            if(Character.isDigit(s.split(" ")[1].toCharArray()[0])){
                res.add(s);
            }else{
                p.offer(s);
            }
        }
        String[] arr = new String[logs.length];
        int count = 0;
        while(p.size()>0){
            arr[count] = p.poll();
            count++;
        }
        int j = 0;
        for(int i = count; i < arr.length; i++ ){
            arr[i]=res.get(j);
            j++;
        }
        return arr;
    }
}