Leetcode2661. 找出叠涂元素

Every day a Leetcode

题目来源：2661. 找出叠涂元素

解法1：哈希

题目很绕，理解题意后就很简单。

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

算法：

特判。
mat 的行数为 m，列数为 n。
建立一个哈希表 unordered_map<int, pair<int, int>> hash，其中 key 是 mat 中整数值，value 是一个 pair<int, int>，存储的是 mat 中 key 值的横坐标、纵坐标。
遍历 mat，其中 key = mat[i][j]，pair<int, int> value(i, j)，插入哈希表 hash 中。
用一个长度为 m 的数组 rowsCount 来表示每一行中已经被涂色的个数，用一个长度为 n 的数组 colsCount 来表示每一列中已经被涂色的个数
遍历数组 arr，设下标为 i，找到 arr[i] 在 mat 中的横纵坐标：row = hash[arr[i]].first，col = hash[arr[i]].second，计数数组对应的行列自增 1，如果发现 rowsCount[row] = n，说明第 row 行的 n 个单元格都被涂上色，返回此时的下标 i；同理，如果发现 colsCount[col] = m，说明第 col 列的 m 个单元格都被涂上色，返回此时的下标 i。

代码：

/** @lc app=leetcode.cn id=2661 lang=cpp** [2661] 找出叠涂元素*/// @lc code=start
class Solution
{
public:int firstCompleteIndex(vector<int> &arr, vector<vector<int>> &mat){if (arr.empty() || mat.empty())return -1;int m = mat.size(), n = m ? mat[0].size() : 0;unordered_map<int, pair<int, int>> hash; // <整数，pair<横坐标，纵坐标>>for (int i = 0; i < m; i++)for (int j = 0; j < n; j++){int key = mat[i][j];pair<int, int> value(i, j);hash[key] = value;}vector<int> rowsCount(m, 0), colsCount(n, 0);for (int i = 0; i < arr.size(); i++){int row = hash[arr[i]].first, col = hash[arr[i]].second;rowsCount[row]++;if (rowsCount[row] == n)return i;colsCount[col]++;if (colsCount[col] == m)return i;}return -1;}
};
// @lc code=end