aboutsummaryrefslogtreecommitdiffstats
path: root/11
diff options
context:
space:
mode:
Diffstat (limited to '')
-rwxr-xr-x1155/main.py31
1 files changed, 31 insertions, 0 deletions
diff --git a/1155/main.py b/1155/main.py
new file mode 100755
index 0000000..c0af065
--- /dev/null
+++ b/1155/main.py
@@ -0,0 +1,31 @@
+#!/usr/bin/env python
+
+
+from functools import lru_cache
+
+
+class Solution:
+ @lru_cache(None)
+ def numRollsToTarget(self, d: int, f: int, target: int) -> int:
+ if d == 1:
+ if 0 < target <= f:
+ return 1
+ return 0
+ if target < d or target > f * d:
+ return 0
+ res = 0
+ for i in range(1, f + 1):
+ res += self.numRollsToTarget(d - 1, f, target - i) % (10**9 + 7)
+ return res % (10**9 + 7)
+
+
+def main():
+ solution = Solution()
+ print(solution.numRollsToTarget(1, 6, 3))
+ print(solution.numRollsToTarget(2, 6, 7))
+ print(solution.numRollsToTarget(2, 5, 10))
+ print(solution.numRollsToTarget(30, 30, 500))
+
+
+if __name__ == "__main__":
+ main()