結果

問題 No.2276 I Want AC
ユーザー umezoumezo
提出日時 2023-04-21 22:37:54
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 71 ms / 2,000 ms
コード長 1,382 bytes
コンパイル時間 2,287 ms
コンパイル使用メモリ 205,304 KB
実行使用メモリ 9,984 KB
最終ジャッジ日時 2024-11-06 16:08:24
合計ジャッジ時間 5,768 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 56
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define ALL(v) v.begin(),v.end()
typedef long long ll;
 
#include<bits/stdc++.h>
using namespace std;

template <typename T>
struct BIT{
  int n;
  vector<T> bit[2];
  BIT(int n_){init(n_);}
  void init(int n_){
    n=n_+1;
    for(int p=0;p < 2;p++) bit[p].assign(n,0);
  }
  
  void add_sub(int p,int i,T x){
    for(int idx=i;idx<n;idx+=(idx & -idx)){
      bit[p][idx] += x;
    }
  }
  
  void add(int l,int r,T x){ //[l,r]に加算
    add_sub(0,l,-x*(l-1));
    add_sub(0,r+1,x*r);
    add_sub(1,l,x);
    add_sub(1,r+1,-x);
  }
  
  T sum_sub(int p,int i){
    T s(0);
    for(int idx=i;idx>0;idx-=(idx & -idx)){
      s+=bit[p][idx];
    }
    return s;
  }
  
  T sum(int i){return sum_sub(0,i)+sum_sub(1,i)*i;}
  
  T sum(int i,int j){  //[i,j]の要素の総和
    return sum(j)-sum(i-1);
  }  
};

int main(){
  ios::sync_with_stdio(false);
  std::cin.tie(nullptr);
  
  int n;
  string s;
  cin>>n>>s;
  
  BIT<ll> bit(n),bit1(n);
  
  ll ans=0;
  ll cnt=0;
  for(int i=n-1;i>=0;i--){
    if(s[i]=='C'){
      cnt++;
    }
    else{
      ans+=cnt;
      bit.add(i+1,i+1,1);
    }
    bit1.add(i+1,i+1,cnt);
  }
  
  ll now=ans;
  for(int i=n-1;i>=0;i--){
    if(s[i]=='?'){
      now-=bit1.sum(i+1,i+1);
      bit1.add(1,i,1);
      now+=bit.sum(1,i);
    }
    ans=max(ans,now);
  }
  cout<<ans<<endl;
    
  return 0;
}
0