結果

問題 No.3118 Increment or Multiply
ユーザー mumumu
提出日時 2025-04-20 18:33:49
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,940 bytes
コンパイル時間 6,378 ms
コンパイル使用メモリ 334,000 KB
実行使用メモリ 814,608 KB
最終ジャッジ日時 2025-04-20 18:33:59
合計ジャッジ時間 8,974 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 1
other AC * 5 WA * 11 MLE * 1 -- * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using mint = modint998244353;
using ll   = long long;
#define rep(i, n) for (ll i = 0; i < (n); i++)
#define reps(i, a, b) for (ll i = (a); i < (b); i++)
bool chmin(auto &a, auto b) { return a > b ? a = b, 1 : 0; }
bool chmax(auto &a, auto b) { return a < b ? a = b, 1 : 0; }
vector<ll> sin45 = {0, 1, 1, 1, 0, -1, -1, -1}, cos45 = {1, 1, 0, -1, -1, -1, 0, 1};
vector<ll> dx = {0, 1, 0, -1};
vector<ll> dy = {1, 0, -1, 0};
#define ft first
#define sd second
#define all(x) std::begin(x), std::end(x)
#define mp(a, b) make_pair(a, b)
#define pii pair<int, int>
#define pll pair<ll, ll>
#define pb(x) push_back(x)
#define so(z) sort(z.begin(), z.end())
#define sor(z) sort(z.rbegin(), z.rend())
#define vec vector<ll>
#define vecc vector<vector<ll>>

int main() {
    cin.tie(nullptr);
    ios_base::sync_with_stdio(false);
    cout << fixed << setprecision(20);

    ll t;
    cin >> t;

    rep(i, t) {
        ll n, a;
        cin >> n >> a;

        mint ans = 0;
        ll k   = 1;
        vector<ll> pow;

        if (a == 1) {
            cout << (mint(n) * mint(n - 1) / mint(2)).val() << endl;
            continue;
        }

        while (k <= n) {
            pow.pb(k);
            k *= a;
        }

        reverse(all(pow));
        ll m = pow.size();

        ll bfr = 1; // どこまで足したか

        for (auto x : pow) {

            ll cnt = n / x - bfr + 1;

            ll aft = (n / x) * x;

            ans += mint(cnt * (m - 1) - (mint(aft) + mint(bfr * mint(x))) * cnt / 2 + mint(n) * mint(cnt));
            m--;

            // cout << "cnt: " << cnt << " m: " << m << " aft: " << aft << " bfr: " << bfr << endl;
            // cout << "x: " << x << " ans: " << ans << endl;
            // cout << "------------------------" << endl;
            bfr += cnt;
        }

        cout << ans.val() << endl;
    }
}
0