fork(1) download
  1. import io, os, sys
  2. input=(io.BytesIO(os.read(0, os.fstat(0).st_size)).readline)
  3. nt=int(input())
  4. nux=0
  5. nin=[]
  6. is_prime=[]
  7. for it in range(nt):
  8. n=int(input())
  9. nin.append(n)
  10. if n>nux: nux=n
  11.  
  12. n=nux+1
  13. sieve = [True]*n
  14. for i in range(3,int(n**0.5)+1,2):
  15. if sieve[i]:
  16. sieve[i*i::2*i]=[False]*((n-i*i-1)//(2*i)+1)
  17. is_prime= [2] + [i for i in range(3,n,2) if sieve[i]]
  18.  
  19. for it in range(nt):
  20. n=nin[it]
  21. if n in is_prime:
  22. sys.stdout.write('TAK\n')
  23. else:
  24. sys.stdout.write('NIE\n')
Success #stdin #stdout 0.03s 9364KB
stdin
5
11
1
5
55
100
stdout
TAK
NIE
TAK
NIE
NIE