题目:
给你一个二维矩阵 matrix 和一个整数 k ,矩阵大小为 m x n 由非负整数组成。
矩阵中坐标 (a, b) 的 值 可由对所有满足 0 <= i <= a < m 且 0 <= j <= b < n 的元素 matrix[i][j](下标从 0 开始计数)执行异或运算得到。
请你找出 matrix 的所有坐标中第 k 大的值(k 的值从 1 开始计数)。
示例 1:
输入:matrix = [[5,2],[1,6]], k = 1
输出:7
解释:坐标 (0,1) 的值是 5 XOR 2 = 7 ,为最大的值。
提示:
m == matrix.lengthn == matrix[i].length1 <= m, n <= 10000 <= matrix[i][j] <= 1061 <= k <= m * n
思路:
我们直接构造出matrix中坐标对应的值,然后将值都放到优先级队列中,
优先级队列中弹出的第K个元素就是答案
代码:
class Solution {
public:
int kthLargestValue(vector<vector<int>>& matrix, int k) {
priority_queue<int> que;
int m = matrix.size(), n = matrix[0].size();
vector<vector<int>> res(m, vector<int>(n));
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (i == 0 && j == 0) {
res[i][j] = matrix[i][j];
}else if (i == 0) {
res[i][j] = res[i][j - 1] ^ matrix[i][j];
}else if (j == 0) {
res[i][j] = res[i - 1][j] ^ matrix[i][j];
}else {
int cur = res[i - 1][j];
for (int k = 0; k <= j; k++) {
cur ^= matrix[i][k];
}
res[i][j] = cur;
}
que.push(res[i][j]);
}
}
while (k > 1) {
que.pop();
k--;
}
return que.top();
}
};