結果

問題 No.1391 ±1 Abs Sum
ユーザー tnakao0123
提出日時 2021-02-16 16:30:48
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,146 bytes
コンパイル時間 825 ms
コンパイル使用メモリ 93,760 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-13 04:01:44
合計ジャッジ時間 4,133 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 13 WA * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

/* -*- coding: utf-8 -*-
 *
 * 1391.cc:  No.1391 ±1 Abs Sum - yukicoder
 */

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<list>
#include<queue>
#include<deque>
#include<algorithm>
#include<numeric>
#include<utility>
#include<complex>
#include<functional>
 
using namespace std;

/* constant */

const int MAX_N = 200000;
const long long LINF = 1LL << 62;

/* typedef */

typedef long long ll;

/* global variables */

int as[MAX_N];

/* subroutines */

ll f(int n, int k, int x) {
  ll sum = 0;
  int r = n - k;
  for (int i = 0, j = n - 1; i <= j;) {
    int di = abs(x - as[i]), dj = abs(x - as[j]);
    if (di >= dj) {
      if (r > 0) sum -= di, r--;
      else sum += di;
      i++;
    }
    else {
      if (r > 0) sum -= dj, r--;
      else sum += dj;
      j--;
    }
  }
  return sum;
}

/* main */

int main() {
  int n, k;
  scanf("%d%d", &n, &k);
  for (int i = 0; i < n; i++) scanf("%d", as + i);

  ll f0 = f(n, k, as[0]);
  ll f1 = f(n, k, as[n - 1]);
  printf("%lld\n", min(f0, f1));
  return 0;
}
0