結果

問題 No.854 公平なりんご分配
ユーザー k
提出日時 2021-03-20 19:33:57
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,400 bytes
コンパイル時間 2,295 ms
コンパイル使用メモリ 216,964 KB
最終ジャッジ日時 2025-01-19 20:25:45
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 56 WA * 35 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

template<typename T>
class SegTree {
  int n;
  vector<T> seg;
  T nil;
  function<T(T,T)> binop;
  
  void update(int k, int l, int r, int p, T x) {
    if (p < l || r <= p) return;
    if (r - l == 1) seg[k] = x;
    else {
      update(2*k+1, l, (l+r)/2, p, x);
      update(2*k+2, (l+r)/2, r, p, x);
      seg[k] = binop(seg[2*k+1], seg[2*k+2]);
    }
  }
  T query(int k, int l, int r, int a, int b) {
    if (b <= l || r <= a) return nil;
    if (a <= l && r <= b) return seg[k];
    return binop(query(2*k+1, l, (l+r)/2, a, b), query(2*k+2, (l+r)/2, r, a, b));
  }
public:
  SegTree(int n, T nil, function<T(T, T)> binop): n(n), seg(4*n), nil(nil), binop(binop) {}
  void init() {
    for (int i = 0; i < 4*n; i++)
      seg[i] = nil;
  }
  // update p-th value to x
  void update(int p, T x) {
    update(0, 0, n, p, x);
  }
  // query for range [l, r)
  T query(int l, int r) {
    return query(0, 0, n, l, r);
  }
};

bool isprime(long long n) {
  if (n < 2) return false;
  if (n == 2) return true;
  if (n % 2 == 0) return false;
  
  for (long long i = 3; i * i <= n; i += 2)
    if (n % i == 0)
      return false;
  
  return true;
}

map<long long, int> factorize(long long n) {
  map<long long, int> ret;
  
  for (long long i = 2; i * i <= n; i++) {
    while (n % i == 0) {
      ++ret[i];
      n /= i;
    }
  }
  
  if (n != 1)
    ret[n] = 1;
  
  return ret;
}

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

  int n;
  cin >> n;
  
  vector<int> index(2000);
  vector<SegTree<short> > trees;

  for (int i = 1; i <= 2000; i++) {
    if (isprime(i)) {
      index[i] = trees.size();
      auto plus = [](short a, short b) { return min(100, a + b); };
      trees.push_back(SegTree<short>(n, 0, plus));      
    }
  }

  for (int i = 0; i < n; i++) {
    int a;
    cin >> a;
    auto fs = factorize(a);
    for (auto &[k, v] :fs) {
      trees[index[k]].update(i, v);
    }
  }

  int q;
  cin >> q;
  for (int i = 0; i < q; i++) {
    int x, l, r;
    cin >> x >> l >> r;

    auto fs = factorize(x);
    bool ck = true;

    for (auto &[k, v]: fs) {
      if (k > 2000) {
        ck = false;
        break;
      }
      if (v > trees[index[k]].query(l-1, r)) {
        ck = false;
        break;
      }
    }

    if (ck)
      cout << "Yes" << endl;
    else
      cout << "NO" << endl;
  }
  

  return 0;
}
0