結果

問題 No.2623 Room Allocation
ユーザー twooimp2
提出日時 2024-02-09 23:17:32
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,914 bytes
コンパイル時間 24,508 ms
コンパイル使用メモリ 357,716 KB
最終ジャッジ日時 2025-02-19 04:13:27
ジャッジサーバーID
（参考情報） 		judge4 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 15 WA * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include<bits/stdc++.h>
#include<atcoder/all>
using namespace std;
using namespace atcoder;
using ll=long long;
using T=tuple<ll,ll,ll>;

void IO(){
  ios::sync_with_stdio(false);
  std::cin.tie(nullptr);
}

int main(){
  IO();
  ll n,x,y;
  cin>>n>>x>>y;
  vector<ll> p(n);
  vector<char> c(n);
  for(ll i=0;i<n;i++){
    cin>>p[i]>>c[i];
  }
  if(n<=x+y){
    ll ans=0;
    for(ll i=0;i<n;i++){
      if(c[i]=='A'){
        if(x>0){
          x--;
          ans+=p[i];
        }
      }else{
        if(y>0){
          y--;
          ans+=p[i];
        }
      }
    }
    cout<<ans<<endl;
  }else{
    vector<ll> sum_a(x+y);
    vector<ll> sum_b(x+y);
    for(ll i=0;i<n;i++){
      if(c[i]=='A'){
        sum_a[i%(x+y)]+=p[i];
      }else{
        sum_b[i%(x+y)]+=p[i];
      }
    }
    vector<T> v;
    for(ll i=0;i<x+y;i++){
      v.emplace_back(sum_a[i],i,0);
      v.emplace_back(sum_b[i],i,1);
    }
    sort(v.rbegin(),v.rend());
    ll rx=x;
    ll ry=y;
    ll ans=0;
    set<ll> fin_idx,second_idx;
    for(ll i=0;i<(ll)v.size();i++){
      ll val=get<0>(v[i]);
      ll idx=get<1>(v[i]);
      ll t=get<2>(v[i]);
      if(fin_idx.find(idx)==fin_idx.end()){
        if(second_idx.find(idx)==second_idx.end()){
          if(t==0&&rx>0){
            rx--;
            fin_idx.insert(idx);
            ans+=val;
          }else if(t==1&&ry>0){
            ry--;
            fin_idx.insert(idx);
            ans+=val;
          }else{
            second_idx.insert(idx);
          }
        }else{
          if(t==0){
            ry--;
            fin_idx.insert(idx);
            second_idx.erase(idx);
            ans+=val;
          }else{
            rx--;
            fin_idx.insert(idx);
            second_idx.erase(idx);
            ans+=val;
          }
        }
      }
    }
    cout<<ans<<endl;
  }
}
0