【教3妹学编程-算法题】找出叠涂元素

3妹：“太阳当空照，花儿对我笑，小鸟说早早早，你为什么背上炸药包”
2哥 :3妹，什么事呀这么开发。
3妹：2哥你看今天的天气多好啊，阳光明媚、万里无云、秋高气爽，适合秋游。
2哥：是啊，立冬之后天气多以多云为主，好不容易艳阳高照。可是你不能秋游，赶紧收拾收拾上班去啦
3妹：哼， 好吧~
2哥：给你出了一道题发你微信里了， 上班通勤的路上记得看一下，回来问你答案~

3妹：知道啦，难不倒我！

题目：

给你一个下标从 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 中的所有整数 互不相同

思路：

由于矩阵 mat中每一个元素都不同，并且都在数组 arr中，所以首先我们用一个哈希表来存储 mat 中每一个元素的位置信息（即行列信息）。然后用一个长度为 n 的数组 rowCnt来表示每一行中已经被涂色的个数，用一个长度为 m 的数组 colCnt来表示每一列中已经被涂色的个数，其中若出现某一行 i 有 rowCnt[i]=m或者某一列 jjj 有 colCnt[j]=n，则表示第 i行或 j 列已被全部涂色。

接着我们遍历数组 arr\textit{arr}arr，对于遍历到的 arr[i]\textit{arr}[i]arr[i]，从哈希表中得到该元素的行列信息，并更新数组 rowCnt\textit{rowCnt}rowCnt 和 colCnt\textit{colCnt}colCnt，如果出现某一行或某一列被全部涂色，则返回该元素的下标 i。

java代码：

class Solution {
    public int firstCompleteIndex(int[] arr, int[][] mat) {
        int m = mat.length, n = mat[0].length;
        int[] rows = new int[m];
        int[] cols = new int[n];
        Map<Integer, Integer> pMap = new HashMap<>();
        for(int i = 0;i<m;i++) {
            for(int j = 0;j<n;j++) {
                int key = i*n+j;
                pMap.put(mat[i][j], key);
            }
        }
        // 开始进行操作
        for(int i = 0;i<arr.length;i++) {
            int num = arr[i];
            int key = pMap.get(num);
            int row = key/n;
            int col = key%n;
            rows[row]++;
            cols[col]++;
            if(rows[row]==n||cols[col]==m) return i;
        }

        return arr.length-1;
    }
}