mickypinata

PROG-T1113: RO

Jun 17th, 2021
644
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. typedef long long lli;
  5.  
  6. map<lli, int> mp;
  7.  
  8. int main(){
  9.  
  10.     int len, k;
  11.     scanf("%d%d", &len, &k);
  12.     int mx = 0;
  13.     lli sum = 0;
  14.     mp[0] = 0;
  15.     for(int i = 1; i <= len; ++i){
  16.         char c;
  17.         scanf(" %c", &c);
  18.         if(c == 'R'){
  19.             sum += k;
  20.         } else if(c == 'O'){
  21.             --sum;
  22.         }
  23.         map<lli, int>::iterator itr = mp.find(sum);
  24.         if(itr == mp.end()){
  25.             mp[sum] = i;
  26.         } else {
  27.             mx = max(mx, i - (itr -> second));
  28.         }
  29.     }
  30.     cout << mx;
  31.  
  32.     return 0;
  33. }
  34.  
RAW Paste Data