結果

問題 No.3118 Increment or Multiply
ユーザー nu
提出日時 2025-04-20 12:20:03
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,113 bytes
コンパイル時間 2,082 ms
コンパイル使用メモリ 197,516 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2025-04-20 12:20:07
合計ジャッジ時間 3,622 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 5 WA * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using u128 = unsigned __int128;

const long long M = 998244353;
long long m2 = (M + 1) / 2;

long long s(long long n, long long a) {
    long long nm = n % M;
    if (a == 1) {
        return nm * ((n - 1) % M) % M * m2 % M;
    }

    vector<long long> b = {1};
    while ((u128)b.back() * a <= n) {
        b.push_back(b.back() * a);
    }
    int L = (int)b.size() - 1;

    long long S = 0, T = 0, U = 1;
    for (int i = 0; i <= L; i++) {
        long long x = n / b[i];
        long long y = (i < L) ? n / b[i + 1] : 0;
        long long k = x - y;
        if (k > 0) {
            S = (S + (long long)i * (k % M)) % M;
            long long z = y + 1;
            long long w = ((z + x) % M * (k % M) % M * m2) % M;
            T = (T + (u128)w * U) % M;
        }
        U = (u128)U * a % M;
    }

    return (S + nm * nm % M - T + M) % M;
}

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

    int T;
    cin >> T;
    while (T--) {
        long long n, a;
        cin >> n >> a;
        cout << s(n, a) << "\n";
    }

    return 0;
}
0