diff --git a/disorderly_escape.py b/disorderly_escape.py index 9ef9c2b..5d5245c 100644 --- a/disorderly_escape.py +++ b/disorderly_escape.py @@ -91,13 +91,16 @@ def cycle_partitions(n, i=1): for p in cycle_partitions(n-i, i): yield [i] + p -def answer(w, h, s): +def solution(w, h, s): grid=0 for cpw in cycle_partitions(w): for cph in cycle_partitions(h): m=cycle_count(cpw, w)*cycle_count(cph, h) grid+=m*(s**sum([sum([gcd(i, j) for i in cpw]) for j in cph])) - return grid//(factorial(w)*factorial(h)) + print grid//(factorial(w)*factorial(h)) -print answer(3, 3, 3) +#print answer(3, 3, 3) + + +#this will surely 100% work by Pratik Ghule