結果

問題 No.3016 ハチマキおじさん
コンテスト
ユーザー のらら
提出日時 2024-12-29 20:51:33
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 432 ms / 2,000 ms
コード長 1,307 bytes
コンパイル時間 5,867 ms
コンパイル使用メモリ 148,332 KB
実行使用メモリ 14,336 KB
最終ジャッジ日時 2025-01-25 21:54:11
合計ジャッジ時間 8,039 ms
ジャッジサーバーID
（参考情報） 		judge6 / judge7
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

int main(){
  ll N;
  cin >> N;
  assert(2 <= N && N <= 200000);
  vector<ll> A(N), B(N - 1);
  for(int i = 0; i < N; i++){
  	cin >> A[i];
  	assert(1 <= A[i] && A[i] <= N);
  }
  for(int i = 0; i < N - 1; i++){
  	cin >> B[i];
  	assert(1 <= B[i] && B[i] <= N);
  }
  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