結果

問題 No.1391 ±1 Abs Sum
ユーザー se1ka2
提出日時 2021-02-12 23:19:51
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,155 bytes
コンパイル時間 709 ms
コンパイル使用メモリ 67,956 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-20 00:59:28
合計ジャッジ時間 3,746 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 15 WA * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

const ll INF = 100000000000000000;

int main()
{
    int n, k;
    cin >> n >> k;
    ll a[200005];
    for(int i = 0; i < n; i++) cin >> a[i];
//    for(int i = n - 1; i >= 0; i--) a[i] -= a[0];
    ll ans = INF;
    int x = 0;
    if(k * 2 >= n) x = (k * 2 - n) / 2;
    if(k * 2 < n){
        for(int c = 0; c < 2; c++){
            ll s = 0;
            reverse(a, a + n);
            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 = min(ans, s);
        }
    }
    else{
        ll s = 0;
        for(int i = 0; i < x; i++) s -= a[i];
        for(int i = x; i < k; i++) s += a[i];
        for(int i = k; i < n; i++) s -= a[i];
        ans = s + a[x] * (x * 2 + n - k * 2);
        for(int i = 0; x + i < n - 1 - x - 2; i += 2){
            if(k + i / 2 >= n || x + i + 2 >= n) break;
            s += (a[i / 2] - a[x + i] - a[x + i + 1] + a[k + i / 2]) * 2;
            ans = min(ans, s + a[x + i + 2] * (x * 2 + n - k * 2 - (i / 2 + 1)));
        }
    }
    cout << ans << endl;
}
0