結果

問題 No.3118 Increment or Multiply
ユーザー cho435
提出日時 2025-04-22 17:19:59
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 923 bytes
コンパイル時間 4,564 ms
コンパイル使用メモリ 258,132 KB
実行使用メモリ 814,464 KB
最終ジャッジ日時 2025-04-22 17:20:10
合計ジャッジ時間 9,821 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 1
other MLE * 1 -- * 34
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using ll = long long;
#define rep(i, s, t) for (ll i = s; i < (ll)(t); i++)
#define all(x) begin(x), end(x)

template <typename T> bool chmin(T& x, T y) {
	return x > y ? (x = y, true) : false;
}
template <typename T> bool chmax(T& x, T y) {
	return x < y ? (x = y, true) : false;
}

struct IOST {
	IOST() {
		ios::sync_with_stdio(false);
		cin.tie(nullptr);
		cout << fixed << setprecision(20);
	}
} IOST;

using mint = atcoder::modint998244353;

map<ll, mint> mp;
mint calc(ll n, ll a) {
	if (n == 1) return 0;
	if (n < a) return mint(n) * (n - 1) / 2;
	if (mp.count(n)) return mp[n];
	ll bn = (n / a);
	return mp[n] = mint(n - bn) * (n - bn - 1) / 2 + calc(bn, a) +
				   bn * (n - bn * a + 1);
}

void solve() {
	ll n, a;
	cin >> n >> a;
	mp.clear();
	cout << calc(n, a).val() << "\n";
}

int main() {
	int t = 1;
	cin >> t;
	rep(i, 0, t) solve();
}
0