From b51c82bb04fce17d9c9ffb06bc74eddf5b4443e3 Mon Sep 17 00:00:00 2001 From: yenru0 Date: Tue, 30 Apr 2024 15:53:18 +0900 Subject: [PATCH] complete 9663_bt.py for pypy3 --- zeta_python/9663_bt.py | 39 ++++++++++++++++++++++++++++ zeta_python/9663_python_optimized.py | 0 2 files changed, 39 insertions(+) create mode 100644 zeta_python/9663_bt.py create mode 100644 zeta_python/9663_python_optimized.py diff --git a/zeta_python/9663_bt.py b/zeta_python/9663_bt.py new file mode 100644 index 0000000..add98b7 --- /dev/null +++ b/zeta_python/9663_bt.py @@ -0,0 +1,39 @@ +import sys + +input = sys.stdin.readline +print = sys.stdout.write + +N = int(input()) + + +def solve(N: int) -> int: + count = 0 + D = [] + for i in range(N): + T = [i] + D.append((1, T)) + + while D: + now, xs = D.pop() + if now == N: + count += 1 + continue + + for i in range(N): + if i in xs: + continue + + flag = True + for j in range(now): + y = xs[j] + if abs(now - j) == abs(y - i): + flag = False + break + if flag: + D.append((now + 1, xs + [i])) + + return count + + +if __name__ == "__main__": + print(str(solve(N))) diff --git a/zeta_python/9663_python_optimized.py b/zeta_python/9663_python_optimized.py new file mode 100644 index 0000000..e69de29