結果

問題 No.2751 429-like Number
ユーザー aradarad
提出日時 2024-05-10 22:39:57
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 158 ms / 4,000 ms
コード長 3,620 bytes
コンパイル時間 5,143 ms
コンパイル使用メモリ 314,280 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-05-10 22:40:05
合計ジャッジ時間 7,847 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 6 ms
6,820 KB
testcase_01 AC 6 ms
6,940 KB
testcase_02 AC 6 ms
6,940 KB
testcase_03 AC 6 ms
6,940 KB
testcase_04 AC 6 ms
6,940 KB
testcase_05 AC 6 ms
6,944 KB
testcase_06 AC 7 ms
6,940 KB
testcase_07 AC 30 ms
6,940 KB
testcase_08 AC 32 ms
6,944 KB
testcase_09 AC 29 ms
6,944 KB
testcase_10 AC 94 ms
6,944 KB
testcase_11 AC 77 ms
6,940 KB
testcase_12 AC 151 ms
6,944 KB
testcase_13 AC 36 ms
6,940 KB
testcase_14 AC 51 ms
6,940 KB
testcase_15 AC 22 ms
6,944 KB
testcase_16 AC 78 ms
6,944 KB
testcase_17 AC 82 ms
6,940 KB
testcase_18 AC 147 ms
6,940 KB
testcase_19 AC 151 ms
6,940 KB
testcase_20 AC 150 ms
6,940 KB
testcase_21 AC 155 ms
6,944 KB
testcase_22 AC 148 ms
6,940 KB
testcase_23 AC 151 ms
6,940 KB
testcase_24 AC 145 ms
6,940 KB
testcase_25 AC 158 ms
6,944 KB
testcase_26 AC 148 ms
6,944 KB
testcase_27 AC 155 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using ll = long long;
using VI = vector<int>;
using VVI = vector<VI>;
using VL = vector<ll>;
using VVL = vector<VL>;
using VD = vector<double>;
using VVD = vector<VD>;
using VS = vector<string>;
using P = pair<ll,ll>;
using VP = vector<P>;
#define rep(i, n) for (ll i = 0; i < ll(n); i++)
#define out(x) cout << x << endl
#define dout(x) cout << fixed << setprecision(10) << x << endl
#define all(a) (a).begin(),(a).end()
#define rall(a) (a).rbegin(),(a).rend()
#define sz(x) (int)(x.size())
#define re0 return 0
#define pcnt __builtin_popcountll
template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; }
template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; }
constexpr int inf = 1e9;
constexpr ll INF = 1e18;
//using mint = modint1000000007;
using mint = modint998244353;
int di[4] = {1,0,-1,0};
int dj[4] = {0,1,0,-1};

// https://youtu.be/UTVg7wzMWQc?t=2774
struct Sieve {
  int n;
  vector<int> f, primes;
  Sieve(int n=1):n(n), f(n+1) {
    f[0] = f[1] = -1;
    for (ll i = 2; i <= n; ++i) {
      if (f[i]) continue;
      primes.push_back(i);
      f[i] = i;
      for (ll j = i*i; j <= n; j += i) {
        if (!f[j]) f[j] = i;
      }
    }
  }
  bool isPrime(int x) { return f[x] == x;}
  vector<int> factorList(int x) {
    vector<int> res;
    while (x != 1) {
      res.push_back(f[x]);
      x /= f[x];
    }
    return res;
  }
  vector<P> factor(int x) {
    vector<int> fl = factorList(x);
    if (fl.size() == 0) return {};
    vector<P> res(1, P(fl[0], 0));
    for (int p : fl) {
      if (res.back().first == p) {
        res.back().second++;
      } else {
        res.emplace_back(p, 1);
      }
    }
    return res;
  }
};

//https://drken1215.hatenablog.com/entry/2023/05/23/233000#chap5
// A^N mod M
template<class T> T pow_mod(T A, T N, T M) {
    T res = 1 % M;
    A %= M;
    while (N) {
        if (N & 1) res = (res * A) % M;
        A = (A * A) % M;
        N >>= 1;
    }
    return res;
}

bool MillerRabin(long long N, vector<long long> A) {
    long long s = 0, d = N - 1;
    while (d % 2 == 0) {
        ++s;
        d >>= 1;
    }
    for (auto a : A) {
        if (N <= a) return true;
        long long t, x = pow_mod<__int128_t>(a, d, N);
        if (x != 1) {
            for (t = 0; t < s; ++t) {
                if (x == N - 1) break;
                x = __int128_t(x) * x % N;
            }
            if (t == s) return false;
        }
    }
    return true;
}

bool is_prime(long long N) {
    if (N <= 1) return false;
    if (N == 2) return true;
    if (N % 2 == 0) return false;
    if (N < 4759123141LL)
        return MillerRabin(N, {2, 7, 61});
    else
        return MillerRabin(N, {2, 325, 9375, 28178, 450775, 9780504, 1795265022});
}

int main(){
    Sieve sieve(5e5);
    VL ps;
    for(ll i = 1;i <= 1e5;i++){
        if(sieve.isPrime(i)) ps.emplace_back(i);
    }
    ll q;
    cin >> q;
    rep(qi,q){
        ll a;
        cin >> a;
        ll cnt = 0;
        for(ll i = 2;i*i*i <= a;i++){
            if(!sieve.isPrime(i)) continue;
            if(a%i==0){
                a /= i;
                cnt++;
                break;
            }
        }
        if(cnt == 0){
            out("No");
            continue;
        }
        bool ok = false;
        rep(i,sz(ps)){
            if(a%ps[i] == 0){
                a /= ps[i];
                ok = true;
                break;
            }
        }
        if(ok && is_prime(a)) out("Yes");
        else out("No");
    }
}
0