結果

問題 No.3106 Simple Math Problem 3
コンテスト
ユーザー shobonvip
提出日時 2025-01-13 20:40:30
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 61 ms / 2,000 ms
コード長 681 bytes
コンパイル時間 1,869 ms
コンパイル使用メモリ 196,568 KB
実行使用メモリ 7,844 KB
最終ジャッジ日時 2025-04-08 22:11:38
合計ジャッジ時間 4,760 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#include<atcoder/modint>
typedef atcoder::modint998244353 mint;
typedef long long ll;

const mint inv2 = mint(2).inv();

mint solve(ll n) {
	mint ans = 0;
	for (ll b=1; b*b<=n; b++) {
		ll num = max(0LL, n - b*b + 1);
		ll q = num / b;
		ll r = num % b;
		ans += mint(b) * (b - 1) * inv2 * q;
		ans += mint(r) * (r - 1) * inv2;
	}

	for (ll k=1; k*k<=n; k++) {
		ll l = k + 1;
		ll r = (n-1) / k;
		if (r - l < 0) continue;
		ans += mint(r - l) * k * (k + 1) * inv2;
		ll lt = r * k + 1;
		ll rt = n;
		ans += mint(rt - lt + 1) * (rt - lt + 2) * inv2;
	}

	return ans;
}

int main() {
	ll n; cin >> n;
	cout << solve(n).val() << '\n';
}
0