自学内容网 自学内容网

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

原文地址:https://blog.csdn.net/weixin_42383726/article/details/145269007

免责声明:本站文章内容转载自网络资源,如侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!