Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def check(n, mod):
- for i in range(mod):
- if i * i % mod == n % mod:
- return 1
- return 0
- def solve(n):
- for it in range(3, 20):
- if not check(n, it):
- return 0
- return 1
- with open("INPUT.TXT") as f:
- n = int(f.readlines()[0])
- res = solve(5 * n * n + 4) or solve(5 * n * n - 4)
- with open("OUTPUT.txt", "w") as f:
- if res:
- f.write("YES\n")
- else:
- f.write("NO\n")
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement