結果

問題 No.2940 Sigma Sigma Div Floor Problem
ユーザー SnowBeenDidingSnowBeenDiding
提出日時 2024-10-18 22:56:26
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 1,380 bytes
コンパイル時間 5,383 ms
コンパイル使用メモリ 312,652 KB
実行使用メモリ 29,984 KB
最終ジャッジ日時 2024-10-18 22:57:18
合計ジャッジ時間 13,423 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 1 ms
6,816 KB
testcase_03 AC 21 ms
6,816 KB
testcase_04 AC 2 ms
6,816 KB
testcase_05 AC 2 ms
6,820 KB
testcase_06 AC 1 ms
6,816 KB
testcase_07 AC 2 ms
6,820 KB
testcase_08 AC 1 ms
6,816 KB
testcase_09 AC 2 ms
6,820 KB
testcase_10 AC 2 ms
6,816 KB
testcase_11 AC 5 ms
6,820 KB
testcase_12 AC 11 ms
6,820 KB
testcase_13 AC 6 ms
6,820 KB
testcase_14 AC 3 ms
6,820 KB
testcase_15 AC 8 ms
6,820 KB
testcase_16 AC 3 ms
6,824 KB
testcase_17 AC 259 ms
6,820 KB
testcase_18 TLE -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
08_evil_01.txt -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <atcoder/all>
#include <bits/stdc++.h>
#define rep(i, a, b) for (ll i = (ll)(a); i < (ll)(b); i++)
using namespace atcoder;
using namespace std;

typedef long long ll;

template <class T, class S = T> struct AllPairOperationSum {
    // https://shakayami.hatenablog.com/entry/2021/01/01/044946
    // Σ op(a[i], a[j]) (i < j)
    ll n;
    vector<T> a;
    AllPairOperationSum() {}
    AllPairOperationSum(vector<T> a) : a(a), n(a.size()) {}
    T floor() {
        // floor(A / B)
        // O(N √maxA log maxA)
        T maxa = *max_element(a.begin(), a.end());
        T sqrtmaxa = sqrt(maxa);
        vector<T> v(sqrtmaxa, 0);
        fenwick_tree<T> ft(maxa + 1);
        T ret = 0;
        rep(i, 0, n) {
            if (a[i] < sqrtmaxa) {
                ret += v[a[i]];
            } else {
                for (T k = 0; k < maxa + 1; k += a[i]) {
                    ret += (k / a[i]) * ft.sum(k, std::min(k + a[i], maxa + 1));
                }
            }
            ft.add(a[i], 1);
            for (T j = 1; j < sqrtmaxa; j++) {
                v[j] += a[i] / j;
            }
        }
        return ret;
    }
};

int main() {
    ll n;
    cin >> n;
    vector<ll> a(n);
    rep(i, 0, n) a[i] = i + 1;
    reverse(a.begin(), a.end());
    AllPairOperationSum<ll> op(a);
    ll ans = op.floor() + n;
    ans %= 998244353;
    cout << ans << endl;
}
0