#1174806 (C++23) No.3607 Sum of Powers of GCDs

• 問題: No.3607 Sum of Powers of GCDs
• ユーザー: pockyny
• 提出日時: 2026-08-02 14:03:52
• 言語: C++23 (gcc 15.2.0 + boost 1.90.0)
• 結果: AC
• 実行時間: 1,497 ms / 2,500 ms
• コード長: 2,989 bytes
• コンパイル時間: 1,654 ms
• コンパイル使用メモリ: 197,544 KB
• 実行使用メモリ: 59,264 KB
• 最終ジャッジ日時: 2026-08-02 14:04:13
• 合計ジャッジ時間: 9,985 ms
• ジャッジサーバーID （参考情報）: judge1_0 / judge3_0

ファイルパターン	結果
sample	AC * 1
other	AC * 11

ソースコード

# pragma GCC target("avx2")
# pragma GCC optimize("O3")
# pragma GCC optimize("unroll-loops")

#include <iostream>
#include <vector>
#include <cmath>
#include <atcoder/modint>

using namespace std;
using namespace atcoder;
using mint = modint998244353;
const int MX = 1000000;
bool isPrime[MX + 10];
mint moe[MX + 10],moe_sum[MX + 10];
// k乗和の累積和 sum[i][j] = 0 ~ i - 1 までのj乗和
mint sum[MX + 10][11];
mint pw(mint a,int x){
    mint ret = 1;
    while(x){
        if(x&1) ret *= a;
        a *= a; x /= 2;
    }
    return ret;
}
mint pre_calc_pow[MX + 10];

typedef long long ll;
// floor(n/i) の種類数は O(√n)
// i <= √nに対して、iで全列挙をする
// floor(n/i) <= √n かつ i > √n に対して、区間の列挙ができる
// 除算回数は約 2√n
template<class F>
inline void quotient_ranges(ll n, F&& f) {
    const ll s = sqrt(n);
    // i <= sqrt(n) を直接列挙
    for (ll i = 1; i <= s; ++i) {
        f(n / i, i, i);
    }
    // i > sqrt(n) を商 q ごとに列挙
    const ll qmax = n / (s + 1);
    ll r = n;
    for (ll q = 1; q <= qmax; ++q) {
        const ll l = max(n / (q + 1) + 1, s + 1);
        f(q, l, r);
        r = l - 1;
    }
}

int main(){
    std::ios_base::sync_with_stdio(false);
    std::cin.tie(nullptr);
    {
        int i,j;
        for(i=2;i<=MX;i++){
            if(isPrime[i]) continue;
            for(j=2*i;j<=MX;j+=i) isPrime[j] = true;
        }
        for(i=2;i<=MX;i++) isPrime[i] = !isPrime[i];
        for(i=1;i<=MX;i++) moe[i] = 1;
        for(i=2;i<=MX;i++){
            if(!isPrime[i]) continue;
            for(j=1;j*i<=MX;j++){
                if(j%i==0) moe[j*i] = 0;
                moe[j*i] *= -1;
            } 
        }
        for(i=1;i<=MX + 1;i++){
            mint val = i - 1;
            for(j=1;j<=10;j++){
                sum[i][j] = sum[i - 1][j] + val;
                val *= (i - 1);
            }
        }
        for(i=1;i<=MX + 1;i++) moe_sum[i] = moe_sum[i - 1] + moe[i - 1];
    }
    int t; cin >> t;
    while(t){
        t--;
        int n,m,k; cin >> n >> m >> k;
        // ans = \sum_i\sum_{floor(m/x) = i}x^k(\sum_{A<=i}moe[A]*floor(i/A)^n)
        // g(i) := (\sum_{A<=i}moe[A]*floor(i/A)^n) として、floor(i/A)が同じ値ごとにまとめて計算する
        // 全体で3/4乗
        quotient_ranges(
            m,
            [&](int q,int l,int r){
                pre_calc_pow[q] = pw(q,n);    
            }
        );
        auto calcG = [&](int i){
            mint ret = 0;
            quotient_ranges(
                i,
                [&](int q,int l,int r){
                    ret += (moe_sum[r + 1] - moe_sum[l])*pre_calc_pow[q];
                }
            );
            return ret;
        };
        mint ans = 0;
        quotient_ranges(
            m,
            [&](int q,int l,int r){
                ans += (sum[r + 1][k] - sum[l][k])*calcG(q);
            }
        );
        cout << ans.val() << "\n";
    }
}