#1187878 (C++23) No.3717 GCD LCM GCD

• 問題: No.3717 GCD LCM GCD
• ユーザー: yaaya
• 提出日時: 2026-09-19 02:42:33
• 言語: C++23 (gcc 15.3.0 + boost 1.92.0 + ACL)
• 結果: AC
• 実行時間: 59 ms / 2,000 ms
• コード長: 4,699 bytes
• コンパイル時間: 2,513 ms
• コンパイル使用メモリ: 353,528 KB
• 実行使用メモリ: 33,092 KB
• 最終ジャッジ日時: 2026-09-19 02:42:43
• 合計ジャッジ時間: 4,240 ms
• ジャッジサーバーID （参考情報）: judge3_1 / judge2_0

ファイルパターン	結果
sample	AC * 4
other	AC * 8

ソースコード

#include <bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(ll i=a;i<b;i++)
#define rrep(i,a,b) for(ll i=a-1;i>=b;i--)
#define ll long long
#define ull unsigned ll
#define ld long double
#define bl __int128_t
#define fi first
#define se second
#define vel vector<ll>
#define vvel vector<vel>
#define pll pair<ll,ll>
#define vepll vector<pll>
#define vvepll vector<vepll>
#define ves vector<string>
#define vem vector<mint>
#define vvem vector<vem>
#define pmm pair<mint,mint>
#define cleout(i) cout<<fixed<<setprecision(i)
template<class T>using PQ=priority_queue<T,vector<T>,greater<T>>;
//               上  右 下 左
vector<int> di={-1, 0, 1, 0};
vector<int> dj={ 0, 1, 0,-1};

vector<int> dx={ 0, 1, 0,-1};
vector<int> dy={ 1, 0,-1, 0};


vector<int> ddx={ 1, 1, 1, 0, -1, -1, -1, 0 };
vector<int> ddy={ 1, 0, -1, -1, -1, 0, 1, 1 };

ll inf=1000000000000000000;//1e18
// LLONG_MAX

mt19937_64 rng((ull)chrono::steady_clock::now().time_since_epoch().count());

struct mint{
    ll num;
    static ll P;
    static void set_mod(ll MOD){
        P=MOD;
    }
    mint(ll x=0){
        if(x<0){
            x*=-1;
            x%=P;
            x=P-x;
        }
        x%=P;
        num=x;
    }
    mint operator+(const mint &other)const{
        return mint(num+other.num);
    }
    mint operator-(const mint &other)const{
        return mint(num-other.num);
    }
    mint operator*(const mint &other)const{
        return mint(num*other.num);
    }
    mint &operator+=(const mint &other){
        num+=other.num;
        if(num>=P) num-=P;
        return *this;
    }
    mint &operator-=(const mint &other){
        num-=other.num;
        if(num<0) num+=P;
        return *this;
    }
    mint &operator*=(const mint &other){
        num=(num*other.num)%P;
        return *this;
    }
    mint beki(const ll &x)const{
        ll pos=x;
        mint res=1;
        mint now=num;
        while(pos){
            if(pos&1){
                res*=now;
            }
            now*=now;
            pos/=2;
        }
        return res;
    }
    mint inv()const{
        mint res=1;
        mint now=num;
        rep(i,0,30){
            if((P-2)&(1ll<<i)){
                res*=now;
            } 
            now*=now;
        }
        return res;
    }
    mint operator/(const mint &other)const{
        return *this*other.inv();
    }
    mint &operator/=(const mint &other){
        num=(num*other.inv())%P;
        return *this;
    }
    operator ll()const{
        return (ll)(num);
    }
    friend ostream& operator<<(ostream& os, const mint& m){
        os << m.num;
        return os;
    }
    friend istream& operator>>(istream& is,mint& m){
        ll x;
        is>>x;
        m=mint(x);
        return is;
    }
};
ll mint::P=998244353;
//ll mint::P=1000000007;

/**
 * @brief 線形篩（素数・最小素因数・メビウス関数の列挙）
 * @details O(N)の計算量で[1,N)の範囲の素数、最小素因数、メビウス関数を同時に計算する。
 *          実行後、グローバル変数の primes に素数リストが、minprime に最小素因数が格納される。
 * @param N 探索する範囲の上限（N未満の値を計算する）
 * @return vel 各インデックスにおけるメビウス関数の値を格納した配列
 */
vel primes;
vel minprime;
vel make_mobius(ll N){
    minprime.assign(N,N);
    vel mobius(N,0);
    mobius[1]=1;
    rep(i,2,N){
        if(minprime[i]>i){
            minprime[i]=i;
            primes.push_back(i);
            mobius[i]=-1;
        }
        for(ll p:primes){
            if(p*i>=N)break;
            minprime[p*i]=p;
            if(minprime[i]==p){//i=p*xとなってる
                mobius[p*i]=0;
                break;
            }else mobius[p*i]=-mobius[i];
        }
    }
    return mobius;
}

void _solve(){
    ll N,K;
    cin>>N>>K;
    vel a(N);
    rep(i,0,N)cin>>a[i];
    make_mobius(1.1e6);
    map<ll,vel> P;
    vector<map<ll,ll>> ma(N);
    rep(i,0,N){
        while(a[i]!=1){
            ll p=minprime[a[i]];
            ll cnt=0;
            while(a[i]%p==0){
                a[i]/=p;
                cnt++;
            }
            P[p].push_back(cnt);
        }
    }
    mint ans=1;
    for(auto it=P.begin();it!=P.end();it++){
        vel v=it->se;
        sort(v.begin(),v.end());
        if(v.size()+N/K<=N){
            //
        }else{
            mint p=it->fi;
            ans*=p.beki(v[N/K-1-(N-v.size())]);
        }
    }
    cout<<ans<<"\n";
}


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

    
    ll _;
    bool multitest=0;
    if(multitest)cin>>_;
    else _=1;
    rep(__,0,_){
        _solve();
    }
}