本文正在参加「Java主题月 - Java 刷题打卡」,详情查看 活动链接
题目描述
给你一个二维矩阵 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 大的值。
提示:
- m == matrix.length
- n == matrix[i].length
- 1 <= m, n <= 1000
- 0 <= matrix[i][j] <= 106
- 1 <= k <= m * n
思路分析
总体思路分为三步
1.先进行每一行的异或操作。
2.再把得到值和同列的上一个元素进行异或。
3.再用ArrayList的排序把k个数输出来。
AC代码
class Solution {
public int kthLargestValue(int[][] matrix, int k) {
int[][] results = new int[matrix.length][matrix[0].length];
for (int i = 0; i < matrix.length; i++) {
results[i][0] = matrix[i][0];
}
for (int i = 0; i < matrix.length; i++) {
for (int j = 1; j < matrix[0].length; j++) {
results[i][j] = results[i][j-1] ^matrix[i][j];
}
}
for (int i = 1; i < matrix.length; i++) {
for (int j = 0; j < matrix[0].length; j++) {
results[i][j] = results[i][j] ^results[i-1][j];
}
}
ArrayList<Integer> integers = new ArrayList<>();
for (int i = 0; i < matrix.length; i++) {
for (int j = 0; j < matrix[0].length; j++) {
integers.add(results[i][j]);
}
}
integers.sort(Integer::compareTo);
return integers.get(integers.size()-k);
}
}
总结
感觉循环结构用得过多,应该还能进行优化。