力扣---6342. 找出叠涂元素

allWu / 2023-05-03 / 原文

给你一个下标从 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 <= 105
1 <= m * n <= 105
1 <= arr[i], mat[r][c] <= m * n
arr 中的所有整数互不相同
mat 中的所有整数互不相同

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/first-completely-painted-row-or-column
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

周赛第二题。

具体思路是利用一个哈希表存储 mat 数组中所有元素的下标。

然后用两个辅助数组，分别表示 mat 中某一行未被涂抹的数量，和某一列未被涂抹的数量。

class Solution {
    public int firstCompleteIndex(int[] arr, int[][] mat) {
        int len1 = mat.length;
        int len2 = mat[0].length;
        Map<Integer, int[]> map = new HashMap<>();
        // 存储 mat 数组中的所有元素的下标。
        for (int i = 0; i < len1; i ++) {
            for (int j = 0; j < len2; j ++) {
                map.put(mat[i][j], new int[]{i, j});
            }
        }
        // 计数
//        第 i 行
        int[] arr1 = new int[len1];
        Arrays.fill(arr1, len2);
//        第 j 列
        int[] arr2 = new int[len2];
        Arrays.fill(arr2, len1);
        for (int i = 0; i < arr.length; i ++) {
            int[] tem = map.get(arr[i]);
            int j = tem[0];
            int k = tem[1];
            mat[j][k] = 0;
            arr1[j] --;
            arr2[k] --;
            // 归 0  表示该行或该列的所有元素都被涂色了。
            if (arr1[j] == 0 || arr2[k] == 0) {
                return i;
            }
        }
        return -1;
    }
}