結果

問題 No.3364 Push_back Operation
コンテスト
ユーザー AK_Mi
提出日時 2025-11-17 23:39:01
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 321 ms / 2,000 ms
コード長 1,354 bytes
コンパイル時間 5,534 ms
コンパイル使用メモリ 334,032 KB
実行使用メモリ 7,844 KB
最終ジャッジ日時 2025-11-17 23:39:13
合計ジャッジ時間 10,969 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 53
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using ll = long long;
using ld = long double;
ll intpow(ll a, ll b) {
    if(b == 1)return a;
    if(b == 0)return 1;

    if(b % 2 == 1){
        ll ret = intpow(a,(b-1)/2);
        ret = (ret*ret)% 998244353;
        return (ret * a)% 998244353;
    }else{
        ll ret = intpow(a,b/2);
        return (ret * ret)% 998244353;
    }
}
ll sums(ll a, ll r){
    if(r == 1){
        return 1;
    }
    if(r % 2 == 1){
        r--;
        ll ret = (sums(a,r/2) * (intpow(a,r/2)+ 1))% 998244353;
        ret += intpow(a,r);
        return ret % 998244353;
    }else{
        ll ret = (sums(a,r/2) * (intpow(a,r/2)+ 1))% 998244353;
        return ret;
    }
        
}

int main(){
    ll n;
    cin >> n;

    ll ans = 0;

    ll p = n;

    for(ll i = 1; i * i < n; i++){
        ll a = ll(n / (i+1)) + 1;
        if(i == 1)ans += p - a + 1;
        else{
            
            ll r = p - a + 1;
            ll pows = intpow(i,a);
            ll wa = sums(i,r);
            ans += (pows * wa)% 998244353;
        }
        ans %= 998244353;
        p = a - 1;
        
    }

    for(ll i = 1; i <= p; i++){
        ll a = n / i;
        a %= 998244353;
        ans += intpow(a,i);
        ans %= 998244353;
    }

    cout << ans << '\n';

    return 0;
}
0