Điểm:
900 (p)
Thời gian:
1.0s
Bộ nhớ:
1023M
Input:
bàn phím
Output:
màn hình
Trong ngày thực tập đầu tiên, thầy Hải 'dới' có một câu đố nho nhỏ cho các học sinh của mình. Cho một số nguyên \(n\), hãy kiểm tra \(n\) có phải là số nguyên tố hay không?
Số nguyên tố là số tự nhiên lớn hơn 1 chỉ có hai ước số dương phân biệt là 1 và chính nó.
Input:
- Gồm một dòng duy nhất là số nguyên \(n (|n| \le 10^{12})\)
Output:
- In ra
YES
nếu \(n\) là số nguyên tố. Ngược lại in raNO
.
Example
Test 1
Input
9
Output
NO
Test 1
Input
7
Output
YES
Bình luận
import sys
import math
def is_prime(n):
if n <= 1:
return False
if n == 2:
return True
if n % 2 == 0:
return False
for i in range(3, int(math.sqrt(n)) + 1, 2):
if n % i == 0:
return False
return True
Đọc số nguyên từ đầu vào
n = int(sys.stdin.read().strip())
Kiểm tra và in kết quả
if is_prime(n):
print("YES")
else:
print("NO")
này mình xem kĩ lắm rồi đúng
ủa bạn, import sys là j vậy bạn