結果

問題 No.1724 [Cherry 3rd Tune A] Lápiz labial de Sonia
コンテスト
ユーザー SSRS
提出日時 2021-10-29 21:27:20
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 175 ms / 2,000 ms
コード長 958 bytes
コンパイル時間 1,702 ms
コンパイル使用メモリ 175,760 KB
実行使用メモリ 7,352 KB
最終ジャッジ日時 2024-10-07 09:49:03
合計ジャッジ時間 11,311 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
int main(){
  int N, K;
  cin >> N >> K;
  vector<int> A(N);
  for (int i = 0; i < N; i++){
    cin >> A[i];
  }
  vector<int> B(N);
  for (int i = 0; i < N; i++){
    cin >> B[i];
  }
  string S;
  int cnt = 0;
  for (int i = 0; i < N; i++){
    if (A[i] > B[i]){
      S += 'A';
      cnt++;
    } else {
      S += 'B';
    }
  }
  if (cnt < K){
    vector<pair<int, int>> P1;
    for (int i = 0; i < N; i++){
      if (S[i] == 'B'){
        P1.push_back(make_pair(B[i] - A[i], i));
      }
    }
    sort(P1.begin(), P1.end());
    for (int i = 0; i < K - cnt; i++){
      S[P1[i].second] = 'A';
    }
  } else if (cnt > K){
    vector<pair<int, int>> P2;
    for (int i = 0; i < N; i++){
      if (S[i] == 'A'){
        P2.push_back(make_pair(A[i] - B[i], i));
      }
    }
    sort(P2.begin(), P2.end());
    for (int i = 0; i < cnt - K; i++){
      S[P2[i].second] = 'B';
    }
  }
  cout << S << endl;
}
0