LeetCode 2661. First Completely Painted Row or Column

🔗 https://leetcode.com/problems/first-completely-painted-row-or-column

题目

• 给一个 m*n 的二维数组，给一个 arr 的一纬数组
• 元素由 [1, m * n] 组成
• 遍历 arr，对二维数组中对应的元素进行染色
• 返回执行到 arr 的第几个 index 的时候，二维数组的某一行或者某一列完成染色

思路

• 建立 hash map，记录某个元素对应的二维数组的下标 x，y
• 遍历 arr，对元素对应的行 x 进行统计，列 y 进行统计，当该行/列的统计值达到 max 时，返回 index

代码

class Solution {
public:int firstCompleteIndex(vector<int>& arr, vector<vector<int>>& mat) {unordered_map<int, pair<int, int>> mp;for (int i = 0; i < mat.size(); i++) {for (int j = 0; j < mat[0].size(); j++) {mp[mat[i][j]] = make_pair(i, j);}}vector<int> row(mat.size()), col(mat[0].size());for (int i = 0; i < arr.size(); i++) {auto pair = mp[arr[i]];row[pair.first]++;col[pair.second]++;if (row[pair.first] == mat[0].size()) return i;if (col[pair.second] == mat.size()) return i;}return 0;}
};