目标
给你一个二维矩阵 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 ，为最大的值。示例 2：
输入：matrix = [[5,2],[1,6]], k = 2
输出：5
解释：坐标 (0,0) 的值是 5 = 5 ，为第 2 大的值。示例 3：
输入：matrix = [[5,2],[1,6]], k = 3
输出：4
解释：坐标 (1,0) 的值是 5 XOR 1 = 4 ，为第 3 大的值。示例 4：
输入：matrix = [[5,2],[1,6]], k = 4
输出：0
解释：坐标 (1,1) 的值是 5 XOR 2 XOR 1 XOR 6 = 0 ，为第 4 大的值。说明：
- m == matrix.length
- n == matrix[i].length
- 1 <= m, n <= 1000
0 <= matrix[i][j] <= 10^6- 1 <= k <= m * n
思路
求二维矩阵 matrix 的第k大的异或坐标值，元素 matrix[i][j] 的异或坐标值等于对matrix[0][0] ~ matrix[i][j]矩阵中的所有值进行异或运算。
我们可以先分别对每一行按列进行异或运算，然后再针对每一列按行进行异或运算。然后将它们放入优先队列，再取第k大的值即可。
官网题解用到了二维前缀和 + 排序/快速选择。// todo
代码
/**
* @date 2024-05-26 19:07
*/
public class KthLargestValue11738 {
public int kthLargestValue(int[][] matrix, int k) {
int m = matrix.length;
int n = matrix[0].length;
PriorityQueue<Integer> q = new PriorityQueue<>((a, b) -> b - a);
int[][] dp = new int[m][n];
for (int i = 0; i < m; i++) {
dp[i][0] = matrix[i][0];
}
for (int i = 0; i < m; i++) {
for (int j = 1; j < n; j++) {
dp[i][j] = dp[i][j - 1] ^ matrix[i][j];
}
}
for (int i = 1; i < m; i++) {
for (int j = 0; j < n; j++) {
dp[i][j] ^= dp[i - 1][j];
q.offer(dp[i][j]);
}
}
for (int i = 0; i < n; i++) {
q.offer(dp[0][i]);
}
int res = 0;
while (k > 0) {
res = q.poll();
k--;
}
return res;
}
}
性能
