LeetCode解法汇总2661. 找出叠涂元素

目录链接:

力扣编程题-解法汇总_分享+记录-CSDN博客

GitHub同步刷题项目:

https://github.com/September26/java-algorithms

原题链接:力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台


描述:

给你一个下标从 0 开始的整数数组 arr 和一个 m x n 的整数 矩阵 mat 。arr 和 mat 都包含范围 [1,m * n] 内的 所有 整数。

从下标 0 开始遍历 arr 中的每个下标 i ,并将包含整数 arr[i] 的 mat 单元格涂色。

请你找出 arr 中在 mat 的某一行或某一列上都被涂色且下标最小的元素,并返回其下标 i 。

示例 1:

image explanation for example 1

输入:arr = [1,3,4,2], mat = [[1,4],[2,3]]
输出:2
解释:遍历如上图所示,arr[2] 在矩阵中的第一行或第二列上都被涂色。

示例 2:

LeetCode解法汇总2661. 找出叠涂元素_第1张图片

输入:arr = [2,8,7,4,1,3,5,6,9], mat = [[3,2,5],[1,4,6],[8,7,9]]
输出:3
解释:遍历如上图所示,arr[3] 在矩阵中的第二列上都被涂色。

提示:

  • m == mat.length
  • n = mat[i].length
  • arr.length == m * n
  • 1 <= m, n <= 105
  • 1 <= m * n <= 105
  • 1 <= arr[i], mat[r][c] <= m * n
  • arr 中的所有整数 互不相同
  • mat 中的所有整数 互不相同

解题思路:

最简单的方式,自然是按照题目的描述,每插入一个数字都检查一次,但是这样时间复杂度比较高。所以我们可以反向思考一下,针对整个矩阵,横向和纵向分别遍历一行或者一列,每次找出最大的那个位置。

每次遍历一行或者一列得出的位置中,找出最小的那个。

代码:

class Solution {
    public int firstCompleteIndex(int[] arr, int[][] mat) {
        Map indexMap = new HashMap<>();
        for (int i = 0; i < arr.length; i++) {
            indexMap.put(arr[i], i);
        }
        int minValue = Integer.MAX_VALUE;
        for (int i = 0; i < mat.length; i++) {
            int itemMinValue = 0;
            for (int j = 0; j < mat[i].length; j++) {
                itemMinValue = Math.max(itemMinValue, indexMap.get(mat[i][j]));
            }
            minValue = Math.min(minValue, itemMinValue);
        }
        System.out.println("minValue:" + minValue);
        for (int j = 0; j < mat[0].length; j++) {
            int itemMinValue = 0;
            for (int i = 0; i < mat.length; i++) {
                itemMinValue = Math.max(itemMinValue, indexMap.get(mat[i][j]));
            }
            minValue = Math.min(minValue, itemMinValue);
        }
        return minValue;
    }
}

你可能感兴趣的:(编程题,leetcode,算法,职场和发展)