2661. 找出叠涂元素 : 常规哈希表运用题

题目描述

这是 LeetCode 上的 2661. 找出叠涂元素 ，难度为中等。

Tag : 「模拟」、「哈希表」、「计数」

给你一个下标从 <math xmlns="http://www.w3.org/1998/Math/MathML"> 0 0 </math>0 开始的整数数组 arr 和一个 <math xmlns="http://www.w3.org/1998/Math/MathML"> m × n m \times n </math>m×n 的整数矩阵 mat。

arr 和 mat 都包含范围 <math xmlns="http://www.w3.org/1998/Math/MathML"> [ 1 ， m × n ] [1，m \times n] </math>[1，m×n] 内的所有整数。

从下标 <math xmlns="http://www.w3.org/1998/Math/MathML"> 0 0 </math>0 开始遍历 arr 中的每个下标 i ，并将包含整数 arr[i] 的 mat 单元格涂色。

请你找出 arr 中在 mat 的某一行或某一列上都被涂色且下标最小的元素，并返回其下标 <math xmlns="http://www.w3.org/1998/Math/MathML"> i i </math>i 。

示例 1：

lua 复制代码

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

输出：2

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

示例 2：

css 复制代码

image explanation for example 2

输入：arr = [2,8,7,4,1,3,5,6,9], mat = [[3,2,5],[1,4,6],[8,7,9]]

输出：3

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

提示：

<math xmlns="http://www.w3.org/1998/Math/MathML"> m = m a t . l e n g t h m = mat.length </math>m=mat.length
<math xmlns="http://www.w3.org/1998/Math/MathML"> n = m a t [ i ] . l e n g t h n = mat[i].length </math>n=mat[i].length
<math xmlns="http://www.w3.org/1998/Math/MathML"> a r r . l e n g t h = m × n arr.length = m \times n </math>arr.length=m×n
<math xmlns="http://www.w3.org/1998/Math/MathML"> 1 < = m , n < = 1 0 5 1 <= m, n <= 10^5 </math>1<=m,n<=105
<math xmlns="http://www.w3.org/1998/Math/MathML"> 1 < = m × n < = 105 1 <= m \times n <= 105 </math>1<=m×n<=105
<math xmlns="http://www.w3.org/1998/Math/MathML"> 1 < = a r r [ i ] , m a t [ r ] [ c ] < = m × n 1 <= arr[i], mat[r][c] <= m \times n </math>1<=arr[i],mat[r][c]<=m×n
arr 中的所有整数互不相同
mat 中的所有整数互不相同

哈希表

利用 mat 的数值各不相同，先使用「哈希表」对 mat 进行转存，以 <math xmlns="http://www.w3.org/1998/Math/MathML"> m a t [ i ] [ j ] mat[i][j] </math>mat[i][j] 为键， <math xmlns="http://www.w3.org/1998/Math/MathML"> ( i , j ) (i, j) </math>(i,j) 为值，方便后续快速查询某个值所在位置。

创建数组 c1 和 c2，分别记录某行某列有多少单元格被涂色，如 c1[x] = a 代表第 <math xmlns="http://www.w3.org/1998/Math/MathML"> x x </math>x 行被涂色单元格数量为 <math xmlns="http://www.w3.org/1998/Math/MathML"> a a </math>a 个，c2[y] = b 代表第 <math xmlns="http://www.w3.org/1998/Math/MathML"> y y </math>y 列被涂色单元格数量为 <math xmlns="http://www.w3.org/1998/Math/MathML"> b b </math>b 个。

遍历所有的 <math xmlns="http://www.w3.org/1998/Math/MathML"> a r r [ i ] arr[i] </math>arr[i]，查询到 <math xmlns="http://www.w3.org/1998/Math/MathML"> a r r [ i ] arr[i] </math>arr[i] 的所在位置 <math xmlns="http://www.w3.org/1998/Math/MathML"> ( x , y ) (x, y) </math>(x,y) 后，更新 c1 和 c2，若某行或某列被完全涂色，返回当前下标。

Java 代码：

Java 复制代码

class Solution {
    public int firstCompleteIndex(int[] arr, int[][] mat) {
        int n = mat.length, m = mat[0].length;
        Map<Integer, int[]> map = new HashMap<>();
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                map.put(mat[i][j], new int[]{i, j});
            }
        }
        int[] c1 = new int[n], c2 = new int[m];
        for (int i = 0; i < n * m; i++) {
            int[] info = map.get(arr[i]);
            int x = info[0], y = info[1];
            if (++c1[x] == m || ++c2[y] == n) return i;
        }
        return -1; // never
    }
}

C++ 代码：

C++ 复制代码

class Solution {
public:
    int firstCompleteIndex(vector<int>& arr, vector<vector<int>>& mat) {
        int n = mat.size(), m = mat[0].size();
        unordered_map<int, pair<int, int>> map;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                map[mat[i][j]] = make_pair(i, j);
            }
        }
        vector<int> c1(n), c2(m);
        for (int i = 0; i < n * m; i++) {
            pair<int, int> info = map[arr[i]];
            int x = info.first, y = info.second;
            if (++c1[x] == m || ++c2[y] == n) return i;
        }
        return -1; // never
    }
};

Python 代码：

Python 复制代码

class Solution:
    def firstCompleteIndex(self, arr: List[int], mat: List[List[int]]) -> int:
        n, m = len(mat), len(mat[0])
        mapping = {mat[i][j]: (i, j) for i in range(n) for j in range(m)}
        c1, c2 = [0] * n, [0] * m
        for i in range(n * m):
            x, y = mapping[arr[i]]
            c1[x], c2[y] = c1[x] + 1, c2[y] + 1
            if c1[x] == m or c2[y] == n: return i
        return -1  # never

TypeScript 代码：

TypeScript 复制代码

function firstCompleteIndex(arr: number[], mat: number[][]): number {
    const n = mat.length, m = mat[0].length;
    const map: { [key: number]: [number, number] } = {};
    for (let i = 0; i < n; i++) {
        for (let j = 0; j < m; j++) {
            map[mat[i][j]] = [i, j];
        }
    }
    const c1 = new Array(n).fill(0), c2 = new Array(m).fill(0);
    for (let i = 0; i < n * m; i++) {
        const [x, y] = map[arr[i]];
        if (++c1[x] == m || ++c2[y] === n) return i;
    }
    return -1; // never
};

时间复杂度： <math xmlns="http://www.w3.org/1998/Math/MathML"> O ( n × m ) O(n \times m) </math>O(n×m)
空间复杂度： <math xmlns="http://www.w3.org/1998/Math/MathML"> O ( n × m ) O(n \times m) </math>O(n×m)

最后

这是我们「刷穿 LeetCode」系列文章的第 No.2661 篇，系列开始于 2021/01/01，截止于起始日 LeetCode 上共有 1916 道题目，部分是有锁题，我们将先把所有不带锁的题目刷完。

在这个系列文章里面，除了讲解解题思路以外，还会尽可能给出最为简洁的代码。如果涉及通解还会相应的代码模板。

为了方便各位同学能够电脑上进行调试和提交代码，我建立了相关的仓库：github.com/SharingSour... 。

在仓库地址里，你可以看到系列文章的题解链接、系列文章的相应代码、LeetCode 原题链接和其他优选题解。

更多更全更热门的「笔试/面试」相关资料可访问排版精美的合集新基地 🎉🎉