結果

問題 No.854 公平なりんご分配
ユーザー kibuna
提出日時 2019-07-26 22:01:16
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 2,417 bytes
コンパイル時間 2,185 ms
コンパイル使用メモリ 180,528 KB
実行使用メモリ 132,864 KB
最終ジャッジ日時 2024-07-02 07:16:23
合計ジャッジ時間 13,901 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 20 WA * 31 RE * 29 TLE * 1 -- * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;
using ll = long long;
using pii = pair<int, int>;
using pll = pair<ll, ll>;
using vi = vector<int>;
using vl = vector<ll>;
using vvi = vector<vi>;
using vvl = vector<vl>;
const ll INF = 1LL << 60;
const ll MOD = 1000000007;
template <class T>
bool chmax(T &a, const T &b) {
    return (a < b) ? (a = b, 1) : 0;
}
template <class T>
bool chmin(T &a, const T &b) {
    return (b < a) ? (a = b, 1) : 0;
}

template <typename T = int>
void primeFactors(T a, map<T, int> &facs) {
    double sqrtA = sqrt(a);
    for (int i = 2; i <= sqrtA + 1e-10; ++i) {
        while (a % i == 0) {
            facs[i]++;
            a /= i;
        }
    }
    if (a > sqrtA)
        facs[a]++;
    return;
}

int main() {
    int n;
    cin >> n;
    vi a(n);
    map<int, int> facs;
    for (int i = 0; i < n; ++i) {
        cin >> a[i];
        primeFactors(a[i], facs);
    }
    int q;
    cin >> q;
    vi p(n), l(n), r(n);
    for (int i = 0; i < q; ++i) {
        cin >> p[i] >> l[i] >> r[i];
        l[i]--;
        r[i]--;
    }

    vi primefacs;
    for (auto &mm : facs) {
        primefacs.push_back(mm.first);
    }
    vvi nprimefacs(n, vi(facs.size(), 0));
    for (int i = 0; i < n; ++i) {
        map<int, int> fac;
        primeFactors(a[i], fac);
        for (auto &mm : fac) {
            int j = lower_bound(primefacs.begin(), primefacs.end(), mm.first) - primefacs.begin();
            nprimefacs[i][j] = mm.second;
        }
    }
    for (int i = 1; i < n; ++i) {
        for (int j = 0; j < facs.size(); ++j) {
            nprimefacs[i][j] += nprimefacs[i - 1][j];
        }
    }
    for (int i = 0; i < q; ++i) {
        vi facsToUse(facs.size(), 0);
        for (int j = 0; j < facs.size(); ++j) {
            facsToUse[j] += nprimefacs[r[i]][j];
            if (l[i] > 0)
                facsToUse[j] -= nprimefacs[l[i]][j];
        }
        map<int, int> pprimes;
        primeFactors(p[i], pprimes);
        bool ok = true;
        for (auto &pp : pprimes) {
            int j = lower_bound(primefacs.begin(), primefacs.end(), pp.first) - primefacs.begin();
            if (j == primefacs.size() || facsToUse[j] < pp.second) {
                ok = false;
                break;
            }
        }
        if (ok)
            cout << "Yes"
                 << "\n";
        else
            cout << "NO"
                 << "\n";
    }

    return 0;
}
0