6342.找出叠涂的元素-343

发布时间 2023-04-30 21:27:19作者: huangxk23

找出叠涂的元素

给你一个下标从 0 开始的整数数组 arr 和一个 m x n 的整数 矩阵 mat 。arr 和 mat 都包含范围 [1,m * n] 内的 所有 整数。

从下标 0 开始遍历 arr 中的每个下标 i ,并将包含整数 arr[i] 的 mat 单元格涂色。

请你找出 arr 中在 mat 的某一行或某一列上都被涂色且下标最小的元素,并返回其下标 i 。

示例 1:

输入:arr = [1,3,4,2], mat = [[1,4],[2,3]]
输出:2
解释:遍历如上图所示,arr[2] 在矩阵中的第一行或第二列上都被涂色。
示例 2:

输入:arr = [2,8,7,4,1,3,5,6,9], mat = [[3,2,5],[1,4,6],[8,7,9]]
输出:3
解释:遍历如上图所示,arr[3] 在矩阵中的第二列上都被涂色。

提示:

m == mat.length
n = mat[i].length
arr.length == m * n
1 <= m, n <= 105
1 <= m * n <= 105
1 <= arr[i], mat[r][c] <= m * n
arr 中的所有整数 互不相同
mat 中的所有整数 互不相同

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/first-completely-painted-row-or-column
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路

阅读理解类题目。

code

class Solution {
public:
    int firstCompleteIndex(vector<int>& arr, vector<vector<int>>& mat) {
        
        int m = mat.size(),n = mat[0].size();
        
        vector<int> row(m,n);
        vector<int> col(n,m);
        
        unordered_map<int,pair<int,int> > pos;
        for(int i = 0;i < m;i ++)
        {
            for(int j = 0;j < n;j ++)
            {
                pos[mat[i][j]] = {i,j};
            }
        }
        
        int i;
        int x,y;
        for(i = 0;i < m * n;i ++)
        {
            x = pos[arr[i]].first;
            y = pos[arr[i]].second;
            row[x] --;if(row[x] == 0) break;
            col[y] --;if(col[y] == 0) break;
        }
        
        return i;
        
    }
};