301224

Author: aluferraz

leetcode/editor/en/Q2466/CountWaysToBuildGoodStrings.py

@@ -0,0 +1,37 @@
+from functools import cache
+from typing import List
+
+# leetcode submit region begin(Prohibit modification and deletion)
+class Solution(object):
+    def countGoodStrings(self, low, high, zero, one):
+        """
+        :type low: int
+        :type high: int
+        :type zero: int
+        :type one: int
+        :rtype: int
+        """
+        MOD = 10 ** 9 + 7
+        @cache
+        def go(size):
+            if size > high:
+                return 0
+            ans = 0
+            if size + zero <= high:
+                if size + zero >= low:
+                    ans += 1
+                ans = (ans + go(size + zero)) % MOD
+            if size + one <= high:
+                if size + one >= low:
+                    ans += 1
+                ans = (ans + go(size + one)) % MOD
+            return ans % MOD
+        return go(0) % MOD
+
+        
+# leetcode submit region end(Prohibit modification and deletion)
+
+
+class CountWaysToBuildGoodStrings(Solution):
+    pass
+    

main.py

@@ -1,11 +1,6 @@
-from leetcode.editor.en.Q1475.FinalPricesWithASpecialDiscountInAShop import FinalPricesWithASpecialDiscountInAShop
-from leetcode.editor.en.Q2182.ConstructStringWithRepeatLimit import ConstructStringWithRepeatLimit
-from leetcode.editor.en.Q2257.CountUnguardedCellsInTheGrid import CountUnguardedCellsInTheGrid
-from leetcode.editor.en.Q2516.TakeKOfEachCharacterFromLeftAndRight import TakeKOfEachCharacterFromLeftAndRight
-from leetcode.editor.en.Q2601.PrimeSubtractionOperation import PrimeSubtractionOperation
-from leetcode.editor.en.Q494.TargetSum import TargetSum
+from leetcode.editor.en.Q2466.CountWaysToBuildGoodStrings import CountWaysToBuildGoodStrings
 from leetcode.editor.en.Q689.MaximumSumOf3NonOverlappingSubarrays import MaximumSumOf3NonOverlappingSubarrays
 
 if __name__ == '__main__':
-    print(MaximumSumOf3NonOverlappingSubarrays().maxSumOfThreeSubarrays([1,2,1,2,6,7,5,1], k = 2))
+    print(CountWaysToBuildGoodStrings().countGoodStrings(low = 3, high = 3, zero = 1, one = 1))