結果

問題 No.1250 汝は倍数なりや?
コンテスト
ユーザー ryhoh
提出日時 2020-11-16 23:31:52
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 3,492 bytes
コンパイル時間 2,018 ms
コンパイル使用メモリ 180,636 KB
実行使用メモリ 703,880 KB
最終ジャッジ日時 2024-07-23 01:50:11
合計ジャッジ時間 5,603 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30 MLE * 1 -- * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;

#define rep(i, n) REP(i, 0, n)
#define REP(i, k, n) for(auto i = k; i != n; i++)
#define rrep(i, n) RREP(i, n, -1)
#define RREP(i, n, k) for(auto i = n; i != k; i--)

#define debug(x) cerr << #x << " " << x << endl
#define print(x) cout << x;
#define println(x) cout << x << endl;
#define spc << " " <<

#define mp make_pair
#define mt make_tuple
#define pf push_front
#define pb push_back
#define ppf pop_front
#define ppb pop_back
#define eb emplace_back
#define np next_permutation

using ll = long long;
using Pii = pair<int, int>;
using Tiii = tuple<int, int, int>;
template<class T>using V = vector<T>;
template<class T>using VV = vector<vector<T>>;
using Vi = vector<int>;
using VVi = vector<vector<int>>;

vector<int> *primes(int size) {
    if (size < 0) throw invalid_argument("prime of minus");

    vector<int> *res = new vector<int>();

    if (size <= 8){
        for (int pr : {2, 3, 5, 7}) {
            if (pr < size) res->push_back(pr);
        }
        return res;
    }

    bool *opened = new bool[size];
    for (int i=0; i<size; ++i) {
        opened[i] = true;
    }

    int limit = (int)sqrt(size);
    assert(3 <= limit);
    for (int i=2; i<limit; ++i) {
        if (opened[i]) {
            res->push_back(i);
            for (int j=i; j<size; j+=i) {
                opened[j] = false;
            }
        }
    }

    for (int i=limit; i<size; ++i) {
        if (opened[i]) {
            res->push_back(i);
        }
    }

    delete[] opened;
    return res;
}

// 素因数分解
map<int, int> *primeFactors(long long N, vector<int> *primes) {
    // check for N
    if (N < 1) throw invalid_argument("invalid N for zero or minus");

    map<int, int> *m = new map<int, int>();

    if (N == 1) {
        m->insert(make_pair(1, 1));
        return m;
    }

    // check for primes
    if ((*primes)[primes->size()-1] < (int)sqrt(N))
        throw invalid_argument("insufficient primes vector");

    bool ok = false;
    while (not ok) {
        bool end_in_mid = false;
        for (int prime : *primes) {
            if (N % prime == 0) {
                (*m)[prime] += 1;
                N /= prime;
                end_in_mid = true;
                break;
            }
            if (prime > N) {
                ok = true;
                end_in_mid = true;
                break;
            }
        }
        if (!end_in_mid) {
          if (N > 1) (*m)[N] = 1;
          break;
        }
    }
    return m;
}

std::vector<int> *P;
std::map<int, int> *base_factors, *add_factors, *H_factors;

int main() {
  cin.tie(0);
  ios::sync_with_stdio(false);

  int N, H;
  cin >> N >> H;
  std::vector<int> A(N+1);
  REP(i, 1, N+1) cin >> A[i];
  REP(i, 1, N+1) {
    if (A[i] < 0) A[i] *= -1;
    else if (A[i] == 0) {
      cout << "YES" << endl;
      return 0;
    }
  }

  P = primes(H+1);
  base_factors = primeFactors(A[1], P);
  REP(i, 2, N+1) {
    add_factors = primeFactors(A[i], P);
    for (auto iter = add_factors->begin(); iter != add_factors->end(); iter++) {
      const int key = iter->first;
      const int val = iter->second;
      (*base_factors)[key] += val;
    }
  }

  H_factors = primeFactors(H, P);
  for (auto iter = H_factors->begin(); iter != H_factors->end(); iter++) {
    const int key = iter->first;
    const int val = iter->second;
    if ((*base_factors)[key] < val) {
      cout << "NO" << endl;
      return 0;
    }
  }

  cout << "YES" << endl;
  return 0;
}
0