Add solution #2661

Author: JoshCrozier

2661-first-completely-painted-row-or-column.js

@@ -0,0 +1,38 @@
+/**
+ * 2661. First Completely Painted Row or Column
+ * https://leetcode.com/problems/first-completely-painted-row-or-column/
+ * Difficulty: Medium
+ *
+ * You are given a 0-indexed integer array arr, and an m x n integer matrix mat. arr and mat
+ * both contain all the integers in the range [1, m * n].
+ *
+ * Go through each index i in arr starting from index 0 and paint the cell in mat containing
+ * the integer arr[i].
+ *
+ * Return the smallest index i at which either a row or a column will be completely painted
+ * in mat.
+ */
+
+/**
+ * @param {number[]} arr
+ * @param {number[][]} mat
+ * @return {number}
+ */
+var firstCompleteIndex = function(arr, mat) {
+  const rows = new Array(mat.length).fill(0);
+  const columns = new Array(mat[0].length).fill(0);
+  const map = new Map();
+
+  for (let i = 0; i < mat.length; i++) {
+    for (let j = 0; j < mat[i].length; j++) {
+      map.set(mat[i][j], [i, j]);
+    }
+  }
+
+  for (let i = 0; i < arr.length; i++) {
+    const [rowIndex, columnIndex] = map.get(arr[i]);
+    if (++rows[rowIndex] === mat[0].length || ++columns[columnIndex] === mat.length) {
+      return i;
+    }
+  }
+};

README.md

@@ -457,6 +457,7 @@
 2649|[Nested Array Generator](./2649-nested-array-generator.js)|Medium|
 2650|[Design Cancellable Function](./2650-design-cancellable-function.js)|Hard|
 2657|[Find the Prefix Common Array of Two Arrays](./2657-find-the-prefix-common-array-of-two-arrays.js)|Medium|
+2661|[First Completely Painted Row or Column](./2661-first-completely-painted-row-or-column.js)|Medium|
 2665|[Counter II](./2665-counter-ii.js)|Easy|
 2666|[Allow One Function Call](./2666-allow-one-function-call.js)|Easy|
 2667|[Create Hello World Function](./2667-create-hello-world-function.js)|Easy|