結果

問題 No.1391 ±1 Abs Sum
コンテスト
ユーザー se1ka2
提出日時 2021-02-12 21:54:49
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 934 bytes
コンパイル時間 631 ms
コンパイル使用メモリ 65,548 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-19 21:11:10
合計ジャッジ時間 3,713 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21 WA * 13
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
using namespace std;
typedef long long ll;

int main()
{
    int n, k;
    cin >> n >> k;
    ll a[200005];
    for(int i = 0; i < n; i++) cin >> a[i];
    ll ans = 0;
    if(k * 2 < n){
        ll s = 0;
        for(int i = 0; i < k; i++) s += a[i] - a[0];
        for(int i = k; i < n; i++) s -= a[i] - a[0];
        ans = s;
        s = 0;
        for(int i = 0; i < n - k; i++) s -= a[n - 1] - a[i];
        for(int i = n - k; i < n; i++) s += a[n - 1] - a[i];
        ans = min(ans, s);
    }
    else{
        int x = (k * 2 - n) / 2;
        ll s = 0;
        for(int i = 0; i < k; i++) s += abs(a[i] - a[x]);
        for(int i = k; i < n; i++) s -= abs(a[i] - a[x]);
        ans = s;
        x = n - 1 - x;
        s = 0;
        for(int i = 0; i < n - k; i++) s -= abs(a[x] - a[i]);
        for(int i = n - k; i < n; i++) s += abs(a[x] - a[i]);
        ans = min(ans, s);
    }
    cout << ans << endl;
}
0