結果

問題 No.3505 Sum of Prod of Root
コンテスト
ユーザー よいちなすの
提出日時 2026-04-18 01:06:18
言語 C++23
(gcc 15.2.0 + boost 1.89.0)
コンパイル:
g++-15 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
TLE  
実行時間 -
コード長 1,632 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,151 ms
コンパイル使用メモリ 341,180 KB
実行使用メモリ 26,916 KB
最終ジャッジ日時 2026-04-18 01:06:51
合計ジャッジ時間 7,574 ms
ジャッジサーバーID
(参考情報)
judge3_0 / judge2_1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample -- * 1
other AC * 3 TLE * 1 -- * 9
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

#include<bits/stdc++.h>

using namespace std;
typedef __int128_t i128;
long long M = 998244353;

long long sr(long long l, long long r) {
    if (l > r) return 0;
    return (long long)(((i128)l + r) * (r - l + 1) / 2 % M);
}

int main() {
    long long N; cin >> N;
    set<long long> E = {1, N + 1};
    for (int k = 3; k <= 60; ++k) {
        for (long long v = 2; ; ++v) {
            i128 x = 1;
            for (int p = 0; p < k; ++p) {
                if (__builtin_mul_overflow(x, v, &x)) { x = N + 1; break; }
            }
            if (x > N) break;
            E.insert((long long)x);
        }
    }
    vector<long long> P(E.begin(), E.end());
    long long ans = 0;
    for (int i = 0; i < P.size() - 1; ++i) {
        long long L = P[i], R = P[i+1] - 1, k3 = 1;
        for (int k = 3; k <= 60; ++k) {
            long long v = pow(L, 1.0/k);
            while (true) {
                i128 x = 1; bool o = 0;
                for(int p=0; p<k; ++p) if(__builtin_mul_overflow(x, v+1, &x)) { o=1; break; }
                if (!o && x <= L) v++; else break;
            }
            while (v > 1) {
                i128 x = 1; for(int p=0; p<k; ++p) x *= v;
                if (x > L) v--; else break;
            }
            if (v <= 1) break;
            k3 = k3 * (v % M) % M;
        }
        for (long long c = L, n, v2; c <= R; c = n + 1) {
            v2 = sqrt(c);
            while ((v2 + 1) * (v2 + 1) <= c) v2++;
            while (v2 * v2 > c) v2--;
            n = min(R, (v2 + 1) * (v2 + 1) - 1);
            ans = (ans + sr(c, n) * (v2 % M) % M * k3) % M;
        }
    }
    cout << ans << endl;
}
0