Hướng dẫn giải của T-prime

Chỉ dùng lời giải này khi không có ý tưởng, và đừng copy-paste code từ lời giải này. Hãy tôn trọng người ra đề và người viết lời giải.
Nộp một lời giải chính thức trước khi tự giải là một hành động có thể bị ban.

Nguồn đề : https://codeforces.com/contest/230/problem/B

Ý tưởng làm bài

  • Ta nhận thấy một số có 3 ước khi số đó là bình phương của một số nguyên tố.
  • Ví dụ: 2 là số nguyên tố và 4 có 3 ước {1, 2, 4}
  • 5 là số nguyên tố và 25 có 3 ước {1, 5, 25}
  • Bài này ta chỉ việc kiểm tra căn bậc 2 của một số có phải là số nguyên tố không, nếu có thì đáp án là YES, ngược lại thì in ra NO
  • (nhớ dùng sàng nguyên tố lưu các số nguyên tố nhỏ hơn hoặc bằng sqrt(1e12)=1e6 nha, không thì lỗi TLE đó - quá thời gian chạy chương trình)

Code

#include <bits/stdc++.h>

using namespace std;

int main() {
   #define int long long

   function <void(vector<int>&, int)>
   getSieve = [&](vector <int>& p, int sz){
      p.resize(sz+1, true);
      p[0]=p[1]=false;

      for(int i=0, maxx=sqrt(sz)+1; i<maxx; ++i){
         if(!p[i]){
            continue;
         }

         for(int j=i*i; j<=sz; j+=i){
            p[j]=false;
         }
      }
   };


   vector <int> p;
   getSieve(p, 1e6);


   int t; cin >> t;

   for(int i=0; i<t; ++i){
      int n; cin >> n;


      int x=sqrt(n);

      if(x*x==n && p[x]){
         cout <<"YES" <<endl;
      }
      else{
         cout <<"NO" <<endl;
      }
   }























   return 0;
}

Bình luận

Hãy đọc nội quy trước khi bình luận.

Không có bình luận tại thời điểm này.