結果

問題 No.1391 ±1 Abs Sum
コンテスト
ユーザー se1ka2
提出日時 2021-02-13 13:27:39
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 91 ms / 2,000 ms
コード長 1,149 bytes
コンパイル時間 581 ms
コンパイル使用メモリ 68,096 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-20 16:19:10
合計ジャッジ時間 3,559 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

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