結果

問題 No.2938 Sigma Sigma Distance Distance Problem
ユーザー GOTKAKO
提出日時 2024-10-18 22:46:04
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 817 bytes
コンパイル時間 2,672 ms
コンパイル使用メモリ 194,848 KB
最終ジャッジ日時 2025-02-24 21:03:11
ジャッジサーバーID
（参考情報） 		judge5 / judge
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    int N; cin >> N;
    vector<int> A(N);
    for(auto &a : A) cin >> a;

    long long answer = 0;
    for(int v=1; v<=500; v++){
        long long sum1 = 0,add1 = 0,sum2 = 0,add2 = 0,sumv = 0,addv = 0;
        for(int i=0; i<N; i++){
            if(A.at(i) < v){
                answer += sumv*(v-A.at(i));
                add1 += v-A.at(i);
            }
            else if(A.at(i) == v){
                answer += sum1+sum2;
                addv++;
            }
            else if(A.at(i) > v){
                answer += sumv*(A.at(i)-v);
                add2 += A.at(i)-v;
            }
            sum1 += add1; sum2 += add2; sumv += addv;
        }
    }
    cout << answer << endl;
}
0