LeetCode 2545. 根据第 K 场考试的分数排序

班里有 m 位学生，共计划组织 n 场考试。给你一个下标从 0 开始、大小为 m x n 的整数矩阵 score ，其中每一行对应一位学生，而 score[i][j] 表示第 i 位学生在第 j 场考试取得的分数。矩阵 score 包含的整数 互不相同 。

另给你一个整数 k 。请你按第 k 场考试分数从高到低完成对这些学生（矩阵中的行）的排序。

返回排序后的矩阵。

示例 1：

输入：score = [[10,6,9,1],[7,5,11,2],[4,8,3,15]], k = 2
输出：[[7,5,11,2],[10,6,9,1],[4,8,3,15]]
解释：在上图中，S 表示学生，E 表示考试。

• 下标为 1 的学生在第 2 场考试取得的分数为 11 ，这是考试的最高分，所以 TA 需要排在第一。
• 下标为 0 的学生在第 2 场考试取得的分数为 9 ，这是考试的第二高分，所以 TA 需要排在第二。
• 下标为 2 的学生在第 2 场考试取得的分数为 3 ，这是考试的最低分，所以 TA 需要排在第三。

m == score.length
n == score[i].length
1 <= m, n <= 250
1 <= score[i][j] <= 105
score 由 不同 的整数组成
0 <= k < n

解法一：直接排序：

class Solution {
public:
    vector<vector<int>> sortTheStudents(vector<vector<int>>& score, int k) {
        set<pair<int, int>, greater<pair<int, int>>> scoreAndId;
        int studentNum = score.size();
        for (int i = 0; i < studentNum; ++i) {
            scoreAndId.insert({score[i][k], i});
        }

        vector<vector<int>> res;
        for (pair<int, int> studentInfo : scoreAndId) {
            res.push_back(score[studentInfo.second]);
        }

        return res;
    }
};

如果有n个学生，此算法时间复杂度为O（nlgn），空间复杂度为O（n）。

解法二：原地排序：

class Solution {
public:
    vector<vector<int>> sortTheStudents(vector<vector<int>>& score, int k) {
        sort(score.begin(), score.end(), [k] (vector<int> &a, vector<int> &b) {
            return a[k] > b[k];
        });
        return score;
    }
};

如果有n个学生，此算法时间复杂度为O（nlgn），空间复杂度为O（1）。