静かで孤独な日記

のんびりたまに

yukicoder No.609 Noelちゃんと星々

  • 中央値に集めるのが最適なんて知らないし、考えもしなかった人は三分探索を思いついたよね。
  • 三分探索知らない人はGoogle先生に問い合わせてみよう。(わかりやすい記事があったはず)
  • 俺スゲー賢いとか思った僕は解説読んで知識増やしていこうね
  • 因みに以下のコードは三分探索の書き方を忘れた人間が苦し紛れに書いたものなので、もっと短くて読みやすいコードを提出一覧から見つける事をオススメします!
#include<bits/stdc++.h>
#define ll long long
using namespace std;

int n;
int a[100001];

inline bool check(ll m1,ll m2){
  ll sum1=0,sum2=0;
  for(int i=0;i<n;i++){
    sum1+=abs(m1-a[i]);
    sum2+=abs(m2-a[i]);
  }
  if(sum1>=sum2)return 1;
  else return 0;
}

int main(){
  cin>>n;
  for(int i=0;i<n;i++){
    cin>>a[i];
  }
  int cont=0;
  ll l=-1000000000,r=1000000000;
  while(cont<80){
    ll mid1=(r+2*l)/3;
    ll mid2=(l+2*r)/3;
    //cerr<<mid1<<" "<<mid2<<endl;
    if(check(mid1,mid2)){
      l=mid1;
    }else{
      r=mid2;
    }
    cont++;
  }
  ll ans=(ll)1e15;
  for(int i=l;i<=r;i++){
    ll nowsum=0;
    for(int j=0;j<n;j++){
      nowsum+=abs(i-a[j]);
    }
    ans=min(ans,nowsum);
  }
  cout<<ans<<endl;
  return 0;
}