結果

問題 No.738 平らな農地
ユーザー しらっ亭
提出日時 2017-05-30 03:42:51
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,137 bytes
コンパイル時間 1,755 ms
コンパイル使用メモリ 172,592 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-13 13:41:07
合計ジャッジ時間 5,079 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 29 WA * 58
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

int64_t solve(const int n, const int k, const vector<int> &A) {
  if (k == 1) return 0;

  int64_t sum = 0;
  __int128_t sum2 = 0;

  double min_var = 1e20;
  int min_var_idx = -1;

  for (int i = 0; i < n; i++) {
    sum += A[i];
    sum2 += 1LL * A[i] * A[i];

    // 範囲外に出たものを取り除く
    if (i >= k) {
      sum -= A[i-k];
      sum2 -= 1LL * A[i-k] * A[i-k];
    }
    if (i >= k - 1) {
      // k 要素の分散を求める
      double avg = 1.0 * sum / k;
      double variance = 1.0 * sum2 / k - avg * avg;
      if (min_var > variance) {
        min_var = variance;
        min_var_idx = i;
      }
    }
  }

  vector<int> B;
  for (int i = 0; i < k; i++) {
    B.emplace_back(A[min_var_idx - i]);
  }

  sort(B.begin(), B.end());

  int median = B[(k+1)/2-1];

  int64_t ans = 0;
  for (int b : B) {
    ans += abs(b - median);
  }

  return ans;
}

int main() {
  cin.tie(nullptr);
  ios::sync_with_stdio(false);

  int n, k; cin >> n >> k;
  vector<int> A(n);
  for (int i = 0; i < n; i++) cin >> A[i];

  cout << solve(n, k, A) << endl;

  return 0;
}
0