21-5-19 找出第 K 大的异或坐标值

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] <= 10^6
• 1 <= k <= m * n

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-kth-largest-xor-coordinate-value
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

解法：二维前缀和+最小堆

这是第二次使用PriorityQueue来维护最小堆了，可以在加深一下自己的印象

public int kthLargestValue(int[][] matrix, int k) {
        int m= matrix.length,n=matrix[0].length;
        //加入哨兵，从[1,1]开始存储
        int[][] sum=new int[m+1][n+1];
        //维护一个大小为k的最小堆
        PriorityQueue<Integer> q = new PriorityQueue<>(k);
        for(int i=1;i<=m;i++){
            for(int j=1;j<=n;j++){
                //因为异或的性质 两次相同的数相于等于0，
                // matrix[i-1][j-1] -1是由于现在遍历的是sum添加了哨兵的原因
                sum[i][j]=sum[i-1][j]^sum[i][j-1]^sum[i-1][j-1]^matrix[i-1][j-1];
                if(q.size()<k){
                    q.add(sum[i][j]);
                }else{
                    if(sum[i][j]>q.peek()){
                        q.poll();
                        q.add(sum[i][j]);
                    }
                }
            }
        }
        return q.peek();
    }

对于Queue 中 PriorityQueue（小根堆 的使用

---

此文章创于本人学习时的记录，如有错误或更优解还请指出