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;
        
    }
};