1738. 找出第 K 大的异或坐标值
难度:中等
来源:力扣(LeetCode)
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
1. 题目:1738. 找出第 K 大的异或坐标值
给你一个二维矩阵 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] <= 106
1 <= k <= m * n
2. 分析
这个题目的描述…我看了n遍才明白题目的意思
已知matrix二维数组,正整数k,且
f(a,b) = matrix[0,0] ^ matrix[0,1] ^ matrix[1,0] ^ matrix[1,1] ^ ... ^ matrix[a,b]
,求数组int[f(a,b)]
的第k大的值是多少?
LeetCode笔记-1738. 找出第 K 大的异或坐标值
根据异或特性,可得出
f(a,b) = f(a-1,b) ^ f(a,b-1) ^ f(a-1,b-1) ^ matrix[a][b]
3. 实现
3.1 二维前缀和 + 排序
3.1.1 Java
public class Solution {
public int kthLargestValue(int[][] matrix, int k) {
// 有几列
int m = matrix.length;
// 有几行
int n = matrix[0].length;
int[] matrixLinear = new int[m * n];
matrixLinear[0] = matrix[0][0];
for (int i = 1; i <= m; i++) {
for (int j = 1; j <= n; j++) {
if (i == 1 && j == 1) {
continue;
} else if (i == 1) {
matrixLinear[j - 1] = matrixLinear[j - 2] ^ matrix[i - 1][j - 1];
} else if (j == 1) {
matrixLinear[(i - 1) * n] = matrixLinear[(i - 2) * n] ^ matrix[i - 1][j - 1];
} else {
matrixLinear[(i - 1) * n + j - 1] = matrixLinear[(i - 2) * n + j - 1] ^ matrixLinear[(i - 1) * n + j - 2] ^ matrixLinear[(i - 2) * n + j - 2] ^ matrix[i - 1][j - 1];
}
}
}
Arrays.sort(matrixLinear);
return matrixLinear[matrixLinear.length - k];
}
}
3.1.2 复杂度分析
- 时间复杂度:O(mn log (mn))。计算二维前缀和的时间复杂度为 O(mn)O,排序的时间复杂度为 O(mn log (mn)),因此总时间复杂度为 O(mn log(mn)))。
- 空间复杂度:O(mn),即为存储二维前缀和需要的空间。