結果

問題 No.2623 Room Allocation
ユーザー twooimp2twooimp2
提出日時 2024-02-09 23:17:32
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,914 bytes
コンパイル時間 7,109 ms
コンパイル使用メモリ 318,068 KB
実行使用メモリ 16,936 KB
最終ジャッジ日時 2024-09-28 16:29:39
合計ジャッジ時間 7,585 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 1 ms
6,820 KB
testcase_02 WA -
testcase_03 AC 2 ms
6,816 KB
testcase_04 AC 2 ms
6,816 KB
testcase_05 AC 2 ms
6,820 KB
testcase_06 WA -
testcase_07 WA -
testcase_08 AC 87 ms
16,932 KB
testcase_09 WA -
testcase_10 AC 1 ms
6,820 KB
testcase_11 AC 2 ms
6,816 KB
testcase_12 AC 2 ms
6,816 KB
testcase_13 AC 2 ms
6,816 KB
testcase_14 AC 26 ms
6,816 KB
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 AC 24 ms
6,820 KB
testcase_28 AC 10 ms
6,820 KB
testcase_29 AC 5 ms
6,816 KB
testcase_30 AC 24 ms
6,820 KB
testcase_31 AC 3 ms
6,820 KB
testcase_32 AC 2 ms
6,820 KB
権限があれば一括ダウンロードができます

ソースコード

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