結果

問題 No.1391 ±1 Abs Sum
ユーザー lorent_kyopro
提出日時 2021-02-13 00:06:10
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 193 ms / 2,000 ms
コード長 836 bytes
コンパイル時間 3,607 ms
コンパイル使用メモリ 196,528 KB
最終ジャッジ日時 2025-01-18 19:57:00
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

int main() {
    size_t n, k;
    cin >> n >> k;
    vector<long long> a(n);
    for (auto& ai : a) cin >> ai;
    if (k == 0) {
        auto calc = [a](long long x) {
            long long res = 0;
            for (auto ai : a) res -= abs(x - ai);
            return res;
        };
        cout << min(calc(a.front()), calc(a.back())) << '\n';
        return 0;
    }
    vector<long long> s(1);
    partial_sum(a.begin(), a.end(), back_inserter(s));
    long long ans = LLONG_MAX;
    size_t l = 0, r = k - 1;
    for (size_t i = 0; i < n; ++i) {
        while (r + 1 < n && a[i] - a[l] > a[r + 1] - a[i]) l++, r++;
        long long now = (n - 2 * l - 2 * r + 2 * i) * a[i] + 2 * (s[l] + s[r + 1] - s[i + 1]) - s[0] - s[n];
        ans = min(ans, now);
    }
    cout << ans << '\n';
}
0