問題

076 – Sum of difference

プログラミング初級者から上級者まで楽しめる、競技プログラミングコンテストサイト「AtCoder」。オンラインで毎週開催プログラミングコンテストを開催しています。競技…

atcoder.jp

記録

• 2021/1/28 Sum of difference Easyの後にしたのでAC。

解法

• ソートして \(A_{1} \leqq … \leqq A_{N}\) として、\(i < j\) に対して \(|A_i - A_j| = A_i - A_j\) となる。
• \(A[i]\) が使われている個数は \(0 – N + 2 \times (i+1)-1\) 個になる。

コード

#include <bits/stdc++.h>
using namespace std;
#include <atcoder/all>
using namespace atcoder;
using ll = long long;
using P = pair<int, int>;
#define rep(i, n) for(int i = 0; i < (n); ++i)
// using mint = modint998244353;
// using mint = modint1000000007;
// const int mod = 1000000007;
// const ll INF = 1LL << 62;
// const int INF = 1001001001;

int main() {
  int N;
  cin >> N;

  vector<ll> A(N);
  rep(i, N) { cin >> A[i]; }

  sort(A.begin(), A.end());

  ll ans = 0;
  rep(i, N) {
    ll cnt = (ll)0 - N + 2 * (i + 1) - 1;
    ans += A[i] * cnt;
  }

  cout << ans << endl;
  return 0;
}