寝癖頭の解法

学習中の覚え書きを投稿、更新していきます。

AtCoder Problems in C++ #B - ピザ

第8回日本情報オリンピック 本選(過去問)から、その提出コードの解答例です。
AtCoderとは、コンテストを通じて、プログラミングやアルゴリズムを学習するサービスです。
atcoder.jp
プログラミングコンテストとは、「与えられた問題をいかに素早く、正確に」解くことができるかを競うものです。
競技プログラミング」を略して、「競プロ」などと呼ばれています。

#B - ピザ

https://www.ioi-jp.org/joi/2008/2009-ho-prob_and_sol/2009-ho.pdf#page=4

僕が作成、提出したコードは、以下のとおりです。

/*
AtCoder Problems in C++
#B - ピザ
https://www.ioi-jp.org/joi/2008/2009-ho-prob_and_sol/2009-ho.pdf#page=4
提出コードの解答例
https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
using ll=long long;
int main(void){
  ll d,n,m;
  cin>>d>>n>>m;
  vector<ll> e(n+1),k(m);
  e[0]=0;
  for(ll i=1;i<n;i++){
    cin>>e[i];
  }
  e[n]=d;
  sort(e.begin(),e.end());
  for(ll i=0;i<m;i++){
    cin>>k[i];
  }
  ll ans=0;
  for(ll i=0;i<m;i++){
    ll j=lower_bound(e.begin(),e.end(),k[i])-e.begin();
    if(j>0){
      ans+=min(k[i]-e[j-1],e[j]-k[i]);
    }
  }
  cout<<ans<<endl;
  return 0;
}