結果

問題 No.3016 ハチマキおじさん
ユーザー のらら
提出日時 2024-12-29 19:49:40
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 215 ms / 2,000 ms
コード長 1,169 bytes
コンパイル時間 3,323 ms
コンパイル使用メモリ 182,216 KB
実行使用メモリ 14,208 KB
最終ジャッジ日時 2025-01-25 22:08:17
合計ジャッジ時間 8,292 ms
ジャッジサーバーID
（参考情報） 		judge6 / judge10
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <atcoder/all>
#include <set>
using namespace std;
using namespace atcoder;
using ll = long long;

int main(){
  ll N;
  cin >> N;
  vector<ll> A(N), B(N - 1);
  for(int i = 0; i < N; i++) cin >> A[i];
  for(int i = 0; i < N - 1; i++) cin >> B[i];
  sort(A.begin(), A.end());
  sort(B.begin(), B.end());
  
  //dpl:小さい方からi個連続で選ぶときの不満度の和、dpr:大きい方からi個連続で選ぶときの不満度の和
  vector<ll> dpl(N, 0), dpr(N, 0);
  for(int i = 0; i < N - 1; i++) dpl[i + 1] = abs(A[i] - B[i]);
  for(int i = 0; i < N - 1; i++) dpr[i + 1] = abs(A[N - 1 - i] - B[N - 2 - i]);
  for(int i = 1; i < N; i++){
    dpl[i] += dpl[i - 1];
    dpr[i] += dpr[i - 1];
  }
  
  //dpl[i] + dpr[N - 1 - i]:i番目のハチマキが手元に残る場合の不満度の和
  ll ans = 1e18;
  for(int i = 0; i < N; i++) ans = min(ans, dpl[i] + dpr[N - 1 - i]);
  set<ll> st;
  for(int i = 0; i < N; i++) if(ans == dpl[i] + dpr[N - 1 - i]) st.insert(A[i]);
  cout << st.size() << endl;
  for(auto itr = st.begin(); itr != st.end(); ++itr) cout << *itr << " ";
  cout << endl;
  return 0;
}
0