Feb 15

Author: siddydutta

2025-02-February-LeetCoding-Challenge/Find the Punishment Number of an Integer.py

@@ -0,0 +1,33 @@
+class Solution:
+    def punishmentNumber(self, n: int) -> int:
+
+        def can_partition(s: str, target: int) -> bool:
+            if not s and target == 0:
+                return True
+            if target < 0:
+                return False
+            for i in range(len(s)):
+                left = int(s[:i+1])
+                right = s[i+1:]
+                if can_partition(right, target - left):
+                    return True
+            return False
+
+        sum_ = 0
+        for num in range(1, n+1):
+            sqr = num * num
+            if can_partition(str(sqr), num):
+                sum_ += sqr
+        return sum_
+
+
+def main():
+    n = 10
+    assert Solution().punishmentNumber(n) == 182
+
+    n = 37
+    assert Solution().punishmentNumber(n) == 1478
+
+
+if __name__ == '__main__':
+    main()

2025-02-February-LeetCoding-Challenge/README.md

@@ -17,8 +17,8 @@
 | February 11 | [1910. Remove All Occurrences of a Substring](https://leetcode.com/problems/remove-all-occurrences-of-a-substring/) | Medium | Solved |
 | February 12 | [2342. Max Sum of a Pair With Equal Sum of Digits](https://leetcode.com/problems/max-sum-of-a-pair-with-equal-sum-of-digits/) | Medium | Solved |
 | February 13 | [3066. Minimum Operations to Exceed Threshold Value II](https://leetcode.com/problems/minimum-operations-to-exceed-threshold-value-ii/) | Medium | Solved |
-| February 14 | [1352. Product of the Last K Numbers](https://leetcode.com/problems/product-of-the-last-k-numbers/) | Medium | Unsolved |
-| February 15 | []() |  |  |
+| February 14 | [1352. Product of the Last K Numbers](https://leetcode.com/problems/product-of-the-last-k-numbers/) | Medium | Solved |
+| February 15 | [2698. Find the Punishment Number of an Integer](https://leetcode.com/problems/find-the-punishment-number-of-an-integer/) | Medium | Solved |
 | February 16 | []() |  |  |
 | February 17 | []() |  |  |
 | February 18 | []() |  |  |
@@ -38,5 +38,5 @@
 | Level  | Problems | Solved | Unsolved |
 | ---    | --- | --- | --- |
 | Easy   | 6 | 6 | 0 |
-| Medium | 7 | 6 | 1 |
+| Medium | 8 | 8 | 0 |
 | Hard   | 0 | 0 | 0 |