1738:找出第 K 大的异或坐标值

src/main/java/leetcode/editor/cn/FindKthLargestXorCoordinateValue.java

@@ -64,26 +64,23 @@ public class FindKthLargestXorCoordinateValue {
             int xLength = matrix.length;
             int yLength = matrix[0].length;
             int[][] result = new int[xLength][yLength];
-            PriorityQueue<Integer> xor=new PriorityQueue<>((o1,o2) -> {return o2-o1;});
+            List<Integer> xor = new ArrayList<>();
+            for (int i = 0; i < xLength; i++) {
+                for (int j = 0; j < yLength; j++) {
+                    if (i == 0 && j == 0) {
                         result[0][0] = matrix[0][0];
-            xor.add(result[0][0]);
-            for (int i = 1; i < xLength; i++) {
-                result[i][0] = result[i - 1][0] ^ matrix[i][0];
-                xor.add(result[i][0]);
-            }
-            for (int i = 1; i < yLength; i++) {
-                result[0][i] = result[0][i - 1] ^ matrix[0][i];
-                xor.add(result[0][i]);
-            }
-            for (int i = 1; i < xLength; i++) {
-                for (int j = 1; j < yLength; j++) {
+                    } else if (i > 0 && j > 0) {
                         result[i][j] = result[i - 1][j] ^ result[i][j - 1] ^ result[i - 1][j - 1] ^ matrix[i][j];
+                    } else if (i == 0) {
+                        result[i][j] = result[i][j - 1] ^ matrix[i][j];
+                    } else {
+                        result[i][j] = result[i - 1][j] ^ matrix[i][j];
+                    }
                     xor.add(result[i][j]);
                 }
-            }for(int i=0;i<k-1;i++){
-                xor.poll();
             }
-            return xor.peek();
+            Collections.sort(xor);
+            return xor.get(xLength * yLength - k);
         }
     }
 //leetcode submit region end(Prohibit modification and deletion)

src/main/java/leetcode/editor/cn/FindKthLargestXorCoordinateValue1.java

@@ -0,0 +1,92 @@
+//给你一个二维矩阵 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 
+// 
+// Related Topics 数组 
+// 👍 23 👎 0
+
+package leetcode.editor.cn;
+
+import java.util.PriorityQueue;
+
+//1738:找出第 K 大的异或坐标值
+public class FindKthLargestXorCoordinateValue1 {
+    public static void main(String[] args) {
+        //测试代码
+        Solution solution = new FindKthLargestXorCoordinateValue1().new Solution();
+        solution.kthLargestValue(new int[][]{{5, 2}, {1, 6}}, 1);
+    }
+
+    //力扣代码
+    //leetcode submit region begin(Prohibit modification and deletion)
+    class Solution {
+        public int kthLargestValue(int[][] matrix, int k) {
+            int xLength = matrix.length;
+            int yLength = matrix[0].length;
+            int[][] result = new int[xLength][yLength];
+            PriorityQueue<Integer> xor = new PriorityQueue<>((o1, o2) -> o2 - o1);
+            result[0][0] = matrix[0][0];
+            xor.add(result[0][0]);
+            for (int i = 1; i < xLength; i++) {
+                result[i][0] = result[i - 1][0] ^ matrix[i][0];
+                xor.add(result[i][0]);
+            }
+            for (int i = 1; i < yLength; i++) {
+                result[0][i] = result[0][i - 1] ^ matrix[0][i];
+                xor.add(result[0][i]);
+            }
+            for (int i = 1; i < xLength; i++) {
+                for (int j = 1; j < yLength; j++) {
+                    result[i][j] = result[i - 1][j] ^ result[i][j - 1] ^ result[i - 1][j - 1] ^ matrix[i][j];
+                    xor.add(result[i][j]);
+                }
+            }
+            for (int i = 0; i < k - 1; i++) {
+                xor.poll();
+            }
+            return xor.peek();
+        }
+    }
+//leetcode submit region end(Prohibit modification and deletion)
+
+}