From 7c60dee83b436e8e8a667bf9f191beced8de4fc3 Mon Sep 17 00:00:00 2001 From: Jack Sangdahl Date: Mon, 10 Mar 2025 00:36:13 -0600 Subject: chore: move cpp and rust code to directory --- other/01_matrix.cpp | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 other/01_matrix.cpp (limited to 'other/01_matrix.cpp') diff --git a/other/01_matrix.cpp b/other/01_matrix.cpp new file mode 100644 index 0000000..f6d9c2d --- /dev/null +++ b/other/01_matrix.cpp @@ -0,0 +1,34 @@ +#include "leetcode.h" + +class Solution { +public: + vector> updateMatrix(vector> &mat) { + int m = mat.size(), n = mat[0].size(); + vector> ans(m, vector(n, INT_MAX)); + queue> qp; + for (int i = 0; i < m; i++) { + for (int j = 0; j < n; j++) { + if (mat[i][j] == 0) { + ans[i][j] = 0; + qp.push({i, j}); + } + } + } + vector> vp{{0, 1}, {1, 0}, {0, -1}, {-1, 0}}; + while (!qp.empty()) { + auto curr = qp.front(); + qp.pop(); + int currX = curr.first, currY = curr.second; + for (auto dirr : vp) { + int x = currX + dirr.first, y = currY + dirr.second; + if (x >= 0 && x < m && y >= 0 && y < n) { + if (ans[x][y] > ans[currX][currY] + 1) { + ans[x][y] = ans[currX][currY] + 1; + qp.push({x, y}); + } + } + } + } + return ans; + } +}; -- cgit v1.2.3