/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 1ms 540.0 KiB
#2 Accepted 1ms 540.0 KiB
#3 Accepted 31ms 540.0 KiB
#4 Accepted 37ms 796.0 KiB
#5 Accepted 51ms 768.0 KiB
#6 Accepted 62ms 1.691 MiB
#7 Accepted 91ms 6.098 MiB
#8 Accepted 33ms 580.0 KiB
#9 Accepted 65ms 616.0 KiB
#10 Accepted 71ms 796.0 KiB
#11 Accepted 80ms 672.0 KiB
#12 Accepted 42ms 2.348 MiB
#13 Accepted 67ms 2.926 MiB
#14 Accepted 58ms 2.891 MiB
#15 Accepted 108ms 7.578 MiB
#16 Accepted 173ms 10.734 MiB
#17 Accepted 110ms 14.723 MiB
#18 Accepted 132ms 15.031 MiB
#19 Accepted 226ms 21.539 MiB
#20 Accepted 106ms 14.531 MiB
#21 Accepted 105ms 14.445 MiB

Code

#include <bits/stdc++.h>
using namespace std;
#define SC               scanf
#define PF               printf
#define ull              unsigned long long
#define ld               long double
#define F                first
#define S                second
#define pb               push_back
#define sort_a(a)        sort(a.begin(),a.end());
#define sort_d(a)        sort(a.rbegin(),a.rend());
#define READ(f)          freopen(f, "r", stdin)
#define WRITE(f)         freopen(f, "w", stdout)
#define rev(s)           reverse(s.begin(),s.end())
#define P(ok)            cout << (ok ? "YES\n": "NO\n")
#define __Heart__              ios_base :: sync_with_stdio(false); cin.tie(NULL);
#define ll long long
typedef pair< ll , ll>                   PII;
void solve()
{
   int n , k , K;
   string s ,SS; cin >> n >> k >> s ; SS = s ; K = k ;
   ll dp[n + 5] = {0} , a = 0 , b = 0;
   map < char , int > mp;
   map < int , int > isQ , prevA ;
   vector < int > v ;
   for(int i = 0 ; i < n ; i++) if(s[i] == '?') isQ[i]++ ;
   int i = n - 1 ;
   while( k > 0 && i >= 0){
     if(s[i] == '?') {
        s[i] = 'B' ;
        k-- ;
     }
     i-- ;
   }
   //---------------------------------
   for(int i = 0 ; i < n ; i++){
    if(s[i] == 'A') a++ ;
    else if(isQ[i]) prevA[i] = a ;
    if(isQ[i] && s[i] == 'B') v.pb(i) ;
   }
   dp[0] = 0;
 //  cout << "bug ==> " <<  s << endl ;
   if(s[0] == 'A') mp['A']++ ;
   if(s[0] == 'B' && isQ[0]) mp['B']++ ;
   if(s[0] == '?') mp['?']++ ;
   for(int i = 1 ; i < n ; i++){
    if(s[i] == 'B'){
        dp[i] = dp[i - 1] + mp['A'] ;
        if(mp['A'] || mp['?'] || isQ[i] || mp['B']) mp['B']++ ;
    }
    else if(s[i] == 'A') mp['A']++ ;
    else if(s[i] == '?') mp['?']++ ;
    if(s[i] != 'B') dp[i] = dp[i - 1] ;
   }
//   for(int i = 0 ; i < n ; i++) cout << dp[i] << " " ; cout << endl ;
  a = 0 ; bool ok = 0 ;
  ll Ans = dp[n - 1] , changeToA = 0 , j = 0 , take = 0 , changA = 0 , cnt = 0;
  for(int i = 0 ; i < n ; i++){
    if(s[i] == '?') {
        ok = 1 ;
    if(j < v.size()){
        ll cur = Ans ;
        ll prevAFound = prevA[v[j]] + changeToA ;
        cur  -= prevAFound ;
        cur += (mp['B'] - 1);
       // cout << i << " " << cur << " " << prevA[v[j]] << " " << Ans << endl ;
        if(cur > Ans){
            Ans = cur ;
            changeToA++ ;
            mp['B']-- ;
            take++ ;
            changA++ ;
            j++ ;
            cnt++ ;
        }
    }
    }
    else if(isQ[i] && cnt < K) {
        ll cur = Ans + mp['B'] - 1 ;
        int in = lower_bound(v.begin() , v.end() , i) - v.begin() ;
        in += changA ;
        if(in >= v.size()) in = v.size() - 1 ;
        ll prevAFound = prevA[v[in]] + changeToA ;
        cur -= prevAFound;
      //  cout << i << " " << " " << mp['B'] << " " << prevAFound << " " << v[in] << " " << cur << endl ;
         if(cur > Ans){
            Ans = cur ;
            cnt++ ;
            take-- ;
            changeToA++;
            ok = 1 ;
            mp['B']-- ;
         }
         else if(take > 0) {
            take-- ;
         } else mp['B']-- ;
    }
    else if(s[i] == 'A') a++ , ok = 1;
    else if(s[i] == 'B' && ok) mp['B']-- ;
   // cout << i << " " << mp['B'] << " " << Ans << endl ;
  }
  cout << Ans << endl ;
}
int main()
{
     __Heart__
   //  READ("0in.txt") ;
   //  WRITE("0out.txt") ;
     int t ; cin >> t ; while(t--) solve() ;
}

Information

Submit By
Type
Submission
Problem
P1110 Subsequence of AB
Language
C++17 (G++ 13.2.0)
Submit At
2024-10-11 23:50:29
Judged At
2024-10-11 23:50:29
Judged By
Score
100
Total Time
226ms
Peak Memory
21.539 MiB