9.13 美团笔试

第五题：

import java.util.*;

public class Main5 {

    public static void main(String[] args){
        Scanner in = new Scanner(System.in);
        while(in.hasNext())
        {
            int n = in.nextInt();
            int[][] data = new int[n][3];
            
            // 第一列是优先级，第二列是必要性，然后输入有n行，每行代表一个任务的优先级和必要性
            // data[i][0]: 优先级，data[i][1]: 必要性，data[i][2]: 第几件事情
            for(int i = 0;i < n;i++)
            {
                for(int j = 0;j < 2;j++)
                {
                    data[i][j] = in.nextInt();
                }
                data[i][2] = i + 1; // 第几件事情
            }
            
            Arrays.sort(data, new Comparator<int[]>() {
                @Override
                public int compare(int[] o1, int[] o2) {
                    // TODO Auto-generated method stub
                    if(o1[1] == o2[1]) return o2[0] - o1[0]; // 必要性相等时，优先级按降序排列
                    return o2[1] - o1[1]; // 必要性不等时，必要性按降序排列
                }
            });
            
            for(int i = 0;i < n;i++) 
            {
                System.out.print(data[i][2] + " ");
            }
        }
    }
}

// 参考: https://www.nowcoder.com/discuss/508743

总结：

二维数组的比较：

Comparator接口(比较器)的Java8 Lambda表达式写法：

1)	Arrays.sort(str, (s1, s2) -> (s1 + s2).compareTo(s2 + s1));

2)	实现Comparator接口的复写compare()方法：
// o1 - o2 是升序排序，o2 - o1 是降序排序。

int people[][] = { { 7, 0 }, { 4, 4 }, { 7, 1 }, { 5, 0 }, { 6, 1 }, { 5, 2 } };
// (H, K): 先按照高度 H (o[0]) 降序排序，再按照 K(o[1]) 个数升序排序。

// 1. 方法1:
Arrays.sort(people, (o1, o2) -> o1[0] == o2[0] ? o1[1] - o2[1]: o2[0] - o1[0]);

// 上面是java8 lambda表达式，意思和下面相同：
// 2. 方法2:
Arrays.sort(people, new Comparator<int[]>() {
    @Override
    public int compare(int[] o1, int[] o2) {
        return o1[0] == o2[0] ? o1[1] - o2[1] : o2[0] - o1[0];
    }
});

// 3. 方法3:
Arrays.sort(people, new Comparator<int[]>() {
    @Override
    public int compare(int[] o1, int[] o2) {
        if(o1[0] == o2[0]) return o1[1] - o2[1];
        return o2[0] - o1[0];
    }
});

参考：

1. 9.13 美团笔试 凉
2. 美团9.13测开笔试
3. 美团9.13笔试
4. 406. 根据身高重建队列