Advertisement
Guest User

Untitled

a guest
May 10th, 2021
338
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.39 KB | None | 0 0
  1. def check(n, mod):
  2. for i in range(mod):
  3. if i * i % mod == n % mod:
  4. return 1
  5. return 0
  6.  
  7. def solve(n):
  8. for it in range(3, 20):
  9. if not check(n, it):
  10. return 0
  11. return 1
  12.  
  13. with open("INPUT.TXT") as f:
  14. n = int(f.readlines()[0])
  15.  
  16. res = solve(5 * n * n + 4) or solve(5 * n * n - 4)
  17.  
  18. with open("OUTPUT.txt", "w") as f:
  19. if res:
  20. f.write("YES\n")
  21. else:
  22. f.write("NO\n")
  23.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement