LeetCode 2661. 找出叠涂元素

LeetCode 2661. 找出叠涂元素

描述

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

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

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

示例 1：

输入：arr = [1,3,4,2], mat = [[1,4],[2,3]]
输出：2

解释：遍历如上图所示，arr[2] 在矩阵中的第一行或第二列上都被涂色。

示例 2：

输入：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 <= 10^5
1 <= m * n <= 10^5
1 <= arr[i], mat[r][c] <= m * n
arr 中的所有整数 互不相同
mat 中的所有整数 互不相同

题解

class Solution {
    public static int firstCompleteIndex(int[] arr, int[][] mat){
        // 用于存储矩阵值及索引信息
        HashMap<Integer,int[]> map = new HashMap<>();
        // 矩阵行列叠涂元素数目信息
        int[] rowArr = new int[mat.length];
        int[] colArr = new int[mat[0].length];
        // 遍历矩阵
        for(int j = 0; j < mat.length; j++){
            for(int k = 0; k < mat[j].length; k++){
                map.put(mat[j][k],new int[]{j,k});
            }
        }
        // 叠涂
        for(int i = 0; i < arr.length; i++){
            int[] index = map.get(arr[i]);
            rowArr[index[0]]++;
            colArr[index[1]]++;
            // 终止
            if(rowArr[index[0]] == colArr.length || colArr[index[1]] == rowArr.length){
                return i;
            }
        }
        return -1;
    }
}