From 51180b2be2e54cadba721acfb8bb4d00193d1ee6 Mon Sep 17 00:00:00 2001
From: Anshullakhera <78374334+Anshullakhera@users.noreply.github.com>
Date: Mon, 21 Oct 2024 22:41:03 +0530
Subject: [PATCH] Create Set Matrix Zero.py

Added Set Matrix Zero.py file
---
 data_structures/arrays/Set Matrix Zero.py | 40 +++++++++++++++++++++++
 1 file changed, 40 insertions(+)
 create mode 100644 data_structures/arrays/Set Matrix Zero.py

diff --git a/data_structures/arrays/Set Matrix Zero.py b/data_structures/arrays/Set Matrix Zero.py
new file mode 100644
index 000000000000..42e8e31dc565
--- /dev/null
+++ b/data_structures/arrays/Set Matrix Zero.py	
@@ -0,0 +1,40 @@
+def setMatrixZeroes(matrix):
+    n = len(matrix)
+    m = len(matrix[0])
+
+    # To store which rows and columns are supposed to be marked with zeroes
+    row = [0] * n
+    col = [0] * m
+
+    # Traverse the matrix using nested loops
+    for i in range(n):
+        for j in range(m):
+            # If the cell contains zero, mark its row and column
+            if matrix[i][j] == 0:
+                row[i] = 1
+                col[j] = 1
+
+    # Update the matrix
+    for i in range(n):
+        for j in range(m):
+            # Set cells to zero if any of the row[i] or col[j] is marked
+            if row[i] or col[j]:
+                matrix[i][j] = 0
+
+    # Print the updated matrix
+    for row in matrix:
+        print(" ".join(map(str, row)))
+
+# Driver Code
+n = int(input("Enter number of rows: "))
+m = int(input("Enter number of columns: "))
+
+# Initialize matrix from user input
+matrix = []
+print("Enter the elements row-wise (space-separated):")
+for i in range(n):
+    row = list(map(int, input().split()))
+    matrix.append(row)
+
+# Function call
+setMatrixZeroes(matrix)