Create 473-Matchsticks-to-Square.py

neetcode-gh · web-flow · commit 9846735ea2e2 · 2022-03-13T19:57:20.000-07:00
diff --git a/473-Matchsticks-to-Square.py b/473-Matchsticks-to-Square.py
@@ -0,0 +1,21 @@
+class Solution:
+    def makesquare(self, matchsticks: List[int]) -> bool:
+        length = sum(matchsticks) // 4
+        sides = [0] * 4
+        
+        if sum(matchsticks) / 4 != length:
+            return False
+        matchsticks.sort(reverse=True)
+        def backtrack(i):
+            if i == len(matchsticks):
+                return True
+            
+            for j in range(4):
+                if sides[j] + matchsticks[i] <= length:
+                    sides[j] += matchsticks[i]
+                    if backtrack(i + 1):
+                        return True
+                    sides[j] -= matchsticks[i]
+            return False
+        
+        return backtrack(0)
