結果

問題 No.837 Noelちゃんと星々2
ユーザー tnakao0123
提出日時 2019-06-15 01:04:47
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
実行時間 -
コード長 1,500 bytes
コンパイル時間 741 ms
コンパイル使用メモリ 87,940 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-11-15 18:45:41
合計ジャッジ時間 3,027 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 5 WA * 24
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:59:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   59 |   scanf("%d", &n);
      |   ~~~~~^~~~~~~~~~
main.cpp:61:36: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   61 |   for (int i = 0; i < n; i++) scanf("%d", ys + i);
      |                               ~~~~~^~~~~~~~~~~~~~

ソースコード

diff # 

/* -*- coding: utf-8 -*-
 *
 * 837.cc:  No.837 Noelちゃんと星々2 - 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 = 100000;
const long long LINF = 1LL << 62;

/* typedef */

typedef long long ll;

/* global variables */

int ys[MAX_N];
ll ysums[MAX_N + 1];

/* subroutines */

ll summove(int i0, int i1, ll av) {
  int k = lower_bound(ys + i0, ys + i1, av) - ys;
  ll sum0 = av * (k - i0) - (ysums[k] - ysums[i0]);
  ll sum1 = (ysums[i1] - ysums[k]) - av * (i1 - k);
  return sum0 + sum1;
}

ll minmove(int i0, int i1) {
  ll av0 = (ysums[i1] - ysums[i0]) / (i1 - i0), av1 = av0 + 1;
  return min(summove(i0, i1, av0), summove(i0, i1, av1));
}

/* main */

int main() {
  int n;
  scanf("%d", &n);

  for (int i = 0; i < n; i++) scanf("%d", ys + i);
  sort(ys, ys + n);

  if (ys[0] == ys[n - 1]) {
    puts("1");
    return 0;
  }

  for (int i = 0; i < n; i++) ysums[i + 1] = ysums[i] + ys[i];

  ll minsum = LINF;
  for (int k = 1; k < n; k++) {
    ll mm0 = minmove(0, k), mm1 = minmove(k, n);
    ll sum =  mm0 + mm1;
    if (minsum > sum) minsum = sum;
    //printf("%d: %lld+%lld=%lld\n", k, mm0, mm1, sum);
  }

  printf("%lld\n", minsum);
  return 0;
}
0