結果

問題 No.2940 Sigma Sigma Div Floor Problem
ユーザー 沙耶花沙耶花
提出日時 2024-10-18 21:48:02
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 6,000 ms
コード長 1,394 bytes
コンパイル時間 5,527 ms
コンパイル使用メモリ 266,308 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-18 22:38:19
合計ジャッジ時間 30,001 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 29 MLE * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <atcoder/all>
#include <bits/stdc++.h>
using namespace std;
using namespace atcoder;
using mint = modint998244353;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf32 1000000001
#define Inf64 1000000000000000000LL
//[[l,r],v] l<=x<=rは, floor(N/x)=v
vector<pair<pair<long long,long long>,long long>> get_floor(long long N){
	vector<pair<pair<long long,long long>,long long>> ret;
	for(long long i=1;i<=N;i++){
		long long v = N/i;
		ret.emplace_back(make_pair(i,0),v);
		i = N/v;
		ret.back().first.second = i;
	}
	
	return ret;
}
mint i2;
mint sum(long long n){
	mint ret = 0;
	ret = n;
	ret *= n+1;
	ret *= i2;
	return ret;
}
mint sum(long long l,long long r){
	if(l==0)return sum(r);
	return sum(r)-sum(l-1);
}
mint i6;
mint f2(long long n){
	mint ret = n;
	ret *= n+1;
	ret *= n*2+1;
	ret *= i6;
	return ret;
}
mint sum2(long long l,long long r){
	if(l==0)return f2(r);
	return f2(r)-f2(l-1);
}

int main(){
	i2 = mint(2).inv();
	i6 = mint(6).inv();
	long long n;
	cin>>n;
	
	auto fs = get_floor(n);
	mint ans= 0;
	rep(i,fs.size()){
		long long L = fs[i].first.first,R = fs[i].first.second;
		long long v = fs[i].second;
		{
			mint tt = n;
			tt *= R-L+1;
			tt *= v;
			ans += tt;
		}
		{
			mint tt = sum(L-1,R-1);
			tt *= v;
			ans -= tt;
		}
		{
			mint tt = sum(v-1);
			tt *= sum(L,R);
			ans -= tt;
		}
	}
	cout<<ans.val()<<endl;
	return 0;
}
0