LeetCode 1738. 找出第 K 大的异或坐标值 | Java 刷题打卡

181 阅读1分钟

本文正在参加「Java主题月 - Java 刷题打卡」,详情查看 活动链接

题目描述

给你一个二维矩阵 matrix 和一个整数 k ,矩阵大小为 m x n 由非负整数组成。

矩阵中坐标 (a, b) 的 值 可由对所有满足0 <= i <= a < m0 <= 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);    
    }
}

总结

感觉循环结构用得过多,应该还能进行优化。