結果
問題 | No.2127 Mod, Sum, Sum, Mod |
ユーザー |
|
提出日時 | 2022-11-18 23:16:05 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 4 ms / 2,000 ms |
コード長 | 1,513 bytes |
コンパイル時間 | 2,011 ms |
コンパイル使用メモリ | 194,636 KB |
最終ジャッジ日時 | 2025-02-08 22:26:41 |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 27 |
ソースコード
#include<bits/stdc++.h>using namespace std;using ll=long long;#define rep(i,n) for(int i=0;i<n;i++)#define rrep(i,n) for(int i=(n)-1;i>=0;i--)#define all(v) v.begin(),v.end()#define rall(v) v.rbegin(),v.rend()template<class T> bool chmax(T &a, T b){if (a < b){a = b;return true;} else return false;}template<class T> bool chmin(T &a, T b){if (a > b){a = b;return true;} else return false;}const ll MOD=998244353;ll f(ll x){return x*(x+1)%MOD*(2*x+1)%MOD*166374059%MOD;}ll g(ll x){return x*(x+1)/2%MOD;}ll n,m;int main(){ios::sync_with_stdio(false);cin.tie(nullptr);cin>>n>>m;ll ans=0;ll x=1;while(x*x<=n&&x<=m){ans+=(x*(x-1)/2%MOD)*(n/x%MOD)%MOD+(n%x)*(n%x+1)/2%MOD;ans%=MOD;x++;}ll y=max(1LL,(n+m-1)/m-10);bool flag=false;while(n/y>=x){if(m>=n/y){ans+=(y*(y+1)/2)%MOD*(f(n/y)-f(n/(y+1)))%MOD;ans%=MOD;ans-=(n+1)*y%MOD*(g(n/y)-g(n/(y+1)))%MOD;ans%=MOD;if(ans<0)ans+=MOD;ans+=n*(n+1)/2%MOD*(n/y-n/(y+1))%MOD;ans%=MOD;}else if(m>n/(y+1)){ans+=(y*(y+1)/2)%MOD*(f(m)-f(n/(y+1)))%MOD;ans%=MOD;ans-=(n+1)*y%MOD*(g(m)-g(n/(y+1)))%MOD;ans%=MOD;if(ans<0)ans+=MOD;ans+=n*(n+1)/2%MOD*(m-n/(y+1))%MOD;ans%=MOD;}y++;}if(m>n){ans+=n*(n+1)/2%MOD*(m-n)%MOD;}cout<<ans<<endl;}