fork download
  1. #include <iostream>
  2. #include <string>
  3. #include <cmath>
  4.  
  5. using namespace std;
  6.  
  7. bool is_prime(long long n) {
  8. if (n <= 1) return false;
  9. if (n == 2) return true;
  10. if (n % 2 == 0) return false;
  11. for (long long i = 3; i * i <= n; i += 2) {
  12. if (n % i == 0) return false;
  13. }
  14. return true;
  15. }
  16.  
  17. void solve() {
  18. long long x, k;
  19. cin >> x >> k;
  20. if (k == 1) {
  21. if (is_prime(x)) {
  22. cout << "YES" << endl;
  23. } else {
  24. cout << "NO" << endl;
  25. }
  26. } else {
  27. cout << "NO" << endl;
  28. }
  29. }
  30.  
  31. int main() {
  32. int t;
  33. cin >> t;
  34. while (t--) {
  35. solve();
  36. }
  37. return 0;
  38. }
Success #stdin #stdout 0s 5284KB
stdin
4
52 3
6 7
7 1
1 7
stdout
NO
NO
YES
NO