結果

問題 No.2938 Sigma Sigma Distance Distance Problem
ユーザー Tatsu_mr
提出日時 2025-03-03 11:47:39
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 1,741 bytes
コンパイル時間 3,892 ms
コンパイル使用メモリ 282,108 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2025-03-03 11:48:12
合計ジャッジ時間 5,915 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#define For(i, a, b) for(int i = (a); i < (b); i++)
#define rep(i, n) For(i, 0, n)
#define rFor(i, a, b) for(int i = (a); i >= (b); i--)
#define ALL(v) (v).begin(), (v).end()
#define rALL(v) (v).rbegin(), (v).rend()

using lint = long long;
using ld = long double;

int INF = 2000000000;
lint LINF = 1000000000000000000;

struct SetupIo {
    SetupIo() {
        ios::sync_with_stdio(false);
        cin.tie(nullptr);
        cout << fixed << setprecision(15);
    }
} setupio;

template <class T>
struct PrefixSum {
    int n;
    vector<T> s;
    
    PrefixSum() {}
    PrefixSum(vector<T> &a) : n(a.size()), s(n + 1, T(0)) {
        for (int i = 0; i < n; i++) {
            s[i + 1] = s[i] + a[i];
        }
    }
   
    // [l, r)
    T prod(int l, int r) {
        assert(0 <= l && l <= r && r <= n);
        return (l == r ? T(0) : s[r] - s[l]);
    }
};

int main() {
    int n;
    cin >> n;
    vector<vector<lint>> id(510);
    rep(i, n) {
        int x;
        cin >> x;
        id[x].emplace_back(i);
    }
    vector<PrefixSum<lint>> sum(510);
    rep(i, 510) {
        sum[i] = PrefixSum<lint>(id[i]);
    }
    lint ans = 0;
    rep(i, 510) {
        if (id[i].size() == 0) { continue; }
        For(j, i + 1, 510) {
            if (id[j].size() == 0) { continue; }
            lint d = j - i;
            rep(k, int(id[i].size())) {
                int b = upper_bound(ALL(id[j]), id[i][k]) - id[j].begin();
                int sz = id[j].size();
                ans += (lint(id[i][k]) * lint(b) - sum[j].prod(0, b)) * d;
                ans += (sum[j].prod(b, sz) - lint(id[i][k]) * lint(sz - b)) * d;
            }
        }
    }
    ans *= 2LL;
    cout << ans << "\n";
}
0