結果

問題 No.837 Noelちゃんと星々2
ユーザー tnakao0123
提出日時 2019-06-15 02:11:20
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 772 ms / 2,000 ms
コード長 1,846 bytes
コンパイル時間 634 ms
コンパイル使用メモリ 86,584 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-11 14:19:34
合計ジャッジ時間 9,553 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 29
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:73:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   73 |   scanf("%d", &n);
      |   ~~~~~^~~~~~~~~~
main.cpp:75:36: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   75 |   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, int av) {
  int k = lower_bound(ys + i0, ys + i1, av) - ys;
  ll sum0 = (ll)av * (k - i0) - (ysums[k] - ysums[i0]);
  ll sum1 = (ysums[i1] - ysums[k]) - (ll)av * (i1 - k);
  return sum0 + sum1;
}

ll minmove(int i0, int i1) {
  int a0 = ys[i0], a1 = ys[i1 - 1];
  while (a0 + 2 < a1) {
    int av0 = ((ll)a0 * 2 + a1) / 3;
    int av1 = (a0 + (ll)a1 * 2) / 3;
    ll sum0 = summove(i0, i1, av0);
    ll sum1 = summove(i0, i1, av1);
    if (sum0 < sum1) a1 = av1;
    else a0 = av0;
  }

  ll minsum = summove(i0, i1, a0);
  for (int i = 1; i <= 2 && ++a0 <= ys[i1 - 1]; i++) {
    ll sum = summove(i0, i1, a0);
    if (minsum > sum) minsum = sum;
  }
  return minsum;
}

/* 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;
  }

  ysums[0] = 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