题目描述
给你一个二维矩阵 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 大的值。
思路
这一题的思路是先用动态规划求出 matrix 的所有坐标的值,然后将这些值排序,再取出第 K 大的值。排序我们可以直接调用 Java 的 API 方法 Arrays.sort()。 如何用动态规划求出 matrix 中每个坐标的值呢?对于动态规划问题,我将其拆解为如下五步曲,这五步都搞清楚了,才能说把动态规划真的掌握了!
- 确定dp数组(dp table)以及下标的含义
- 确定递推公式
- dp数组如何初始化
- 确定遍历顺序
- 举例推导dp数组
下面我们就用这五部曲实践一下。
-
确定dp数组以及下标的含义
我们要用一个二维 dp 数组来保存递归的结果,dp[i][j] 的含义为 matrix 的坐标 (i, j) 的值 -
确定递推公式
每个坐标 (i, j) 的值等于其左上角坐标的值,左边坐标的值,上边坐标的值,以及 matrix[i][j] 相互异或运算得到的结果,即dp[i][j] = dp[i-1][j-1] ^ dp[i-1][j] ^ dp[i][j-1] ^ matrix[i][j]
-
dp数组如何初始化
坐标 (0, 0) 的值即为 matrix[0][0] 的值,即dp[0][0] = matrix[0][0];
-
确定遍历顺序
从递归公式dp[i][j] = dp[i-1][j-1] ^ dp[i-1][j] ^ dp[i][j-1] ^ matrix[i][j]
中可以看出,dp[i][j] 是依赖 dp[i - 1][j - 1] 和 dp[i - 1][j] 和 dp[i][j - 1],那么遍历的顺序一定是从前到后遍历的 -
举例推导dp数组
按照这个递推公式dp[i][j] = dp[i-1][j-1] ^ dp[i-1][j] ^ dp[i][j-1] ^ matrix[i][j]
,我们来推导一下,当 matrix = [[5,2],[1,6]] 的时候,dp数组应该是如下的数组:
[[5, 7],[4, 0]]
如果代码写出来,发现结果不对,就把dp数组打印出来看看和我们推导的数列是不是一致的。
Java代码
class Solution {
public int kthLargestValue(int[][] matrix, int k) {
int m = matrix.length;
int n = matrix[0].length;
int[]