結果

問題 No.2896 Monotonic Prime Factors
ユーザー hiro71687khiro71687k
提出日時 2024-09-21 03:30:03
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 929 ms / 2,000 ms
コード長 3,476 bytes
コンパイル時間 4,530 ms
コンパイル使用メモリ 266,884 KB
実行使用メモリ 128,812 KB
最終ジャッジ日時 2024-09-21 03:30:36
合計ジャッジ時間 21,953 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 679 ms
126,132 KB
testcase_01 AC 655 ms
126,024 KB
testcase_02 AC 671 ms
126,040 KB
testcase_03 AC 698 ms
125,968 KB
testcase_04 AC 876 ms
128,812 KB
testcase_05 AC 925 ms
128,808 KB
testcase_06 AC 892 ms
128,812 KB
testcase_07 AC 858 ms
128,684 KB
testcase_08 AC 852 ms
128,652 KB
testcase_09 AC 929 ms
128,812 KB
testcase_10 AC 784 ms
127,420 KB
testcase_11 AC 699 ms
126,336 KB
testcase_12 AC 687 ms
126,176 KB
testcase_13 AC 813 ms
127,544 KB
testcase_14 AC 816 ms
128,376 KB
testcase_15 AC 695 ms
126,108 KB
testcase_16 AC 745 ms
126,744 KB
testcase_17 AC 684 ms
126,332 KB
testcase_18 AC 891 ms
128,692 KB
testcase_19 AC 701 ms
126,464 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include<atcoder/all>
using namespace atcoder;
using namespace std;
using ll=long long;
using ld=long double;
ld pie=3.141592653589793;
ll mod=998244353;
ll inf=10000000000000000;
const int MAX = 5100000;
const int MOD = mod;

long long fac[MAX], finv[MAX], inv[MAX];

// テーブルを作る前処理
void COMinit() {
    fac[0] = fac[1] = 1;
    finv[0] = finv[1] = 1;
    inv[1] = 1;
    for (int i = 2; i < MAX; i++){
        fac[i] = fac[i - 1] * i % MOD;
        inv[i] = MOD - inv[MOD%i] * (MOD / i) % MOD;
        finv[i] = finv[i - 1] * inv[i] % MOD;
    }
}

// 二項係数計算
long long COM(ll n, ll k){
    if (n < k) return 0;
    if (n < 0 || k < 0) return 0;
    return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD;
}
struct Eratosthenes {
    // テーブル
    vector<bool> isprime;

    // 整数 i を割り切る最小の素数
    vector<ll> minfactor;

    vector<ll>mobius;

    // コンストラクタで篩を回す
    Eratosthenes(ll N) : isprime(N+1, true),
                          minfactor(N+1, -1),
                          mobius(N+1,1) {
        // 1 は予めふるい落としておく
        isprime[1] = false;
        minfactor[1] = 1;

        // 篩
        for (ll p = 2; p <= N; ++p) {
            // すでに合成数であるものはスキップする
            if (!isprime[p]) continue;

            // p についての情報更新
            minfactor[p] = p;
            mobius[p]=-1;

            // p 以外の p の倍数から素数ラベルを剥奪
            for (ll q = p * 2; q <= N; q += p) {
                // q は合成数なのでふるい落とす
                isprime[q] = false;

                // q は p で割り切れる旨を更新
                if (minfactor[q] == -1) minfactor[q] = p;
                if ((q / p) % p == 0) mobius[q] = 0;
                else mobius[q] = -mobius[q];
            }
        }
    }

    // 高速素因数分解
    // pair (素因子, 指数) の vector を返す
    vector<pair<ll,ll>> factorize(ll n) {
        vector<pair<ll,ll>> res;
        while (n > 1) {
            ll p = minfactor[n];
            ll exp = 0;

            // n で割り切れる限り割る
            while (minfactor[n] == p) {
                n /= p;
                ++exp;
            }
            res.emplace_back(p, exp);
        }
        return res;
    }
    vector<ll>divisors(ll n){
        vector<ll>res({1});
        auto pf=factorize(n);
        for (auto p : pf)
        {
            ll s=(ll)res.size();
            for (ll i = 0; i < s; i++)
            {
                ll v=1;
                for (ll j = 0; j < p.second; j++)
                {
                    v*=p.first;
                    res.push_back(res[i]*v);
                }
                
            }
            
        }
        return res;
    }  
};
int main(){
    ll q;
    cin >> q;
    vector<ll>a(q),b(q);
    ll x=1;
    Eratosthenes er(200000);
    ll now=0;
    vector<ll>ans;
    COMinit();
    for (ll i = 0; i < q; i++)
    {
        cin >> a[i] >> b[i];
        vector<pair<ll,ll>>p=er.factorize(a[i]);
        for (ll j = 0; j < p.size(); j++)
        {
            now+=p[j].second;
        }
        if (now<b[i])
        {
            ans.push_back(0);
        }else{
            ans.push_back(COM(now-b[i]+b[i]-1,b[i]-1));
        }
    }
    for (ll i = 0; i < ans.size(); i++)
    {
        cout << ans[i] << endl;
    }
    
}
0