結果

問題 No.2554 MMA文字列2 (Query Version)
ユーザー Nzt3
提出日時 2024-01-12 00:35:30
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 460 ms / 5,000 ms
コード長 1,535 bytes
コンパイル時間 2,528 ms
コンパイル使用メモリ 198,316 KB
最終ジャッジ日時 2025-02-18 17:22:27
ジャッジサーバーID
(参考情報)
judge4 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 55
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
#include<atcoder/segtree>
using namespace std;
using ll=long long;
constexpr int CHAR_NUM=26;

struct S{
  array<ll,CHAR_NUM>MA,A;
  ll MMA;
  void print(){
    cout<<MMA<<'/';
    for(int i=0;i<26;i++){
      cout<<A[i]<<',';
    }cout<<'/';
    for(int i=0;i<26;i++){
      cout<<MA[i]<<',';
    }cout<<'\n';
  }
};
S seg_e(){
  S ret;
  for(int i=0;i<CHAR_NUM;i++){
    ret.MA[i]=0;
    ret.A[i]=0;
  }
  ret.MMA=0;
  return ret;
}
S seg_op(S a,S b){
  S ret=seg_e();
  ret.MMA=a.MMA+b.MMA;
  for(int i=0;i<CHAR_NUM;i++){
    ret.A[i]+=a.A[i]+b.A[i];
    ret.MA[i]+=a.MA[i]+b.MA[i];
  }
  ll aMA_s=accumulate(a.MA.begin(),a.MA.end(),0ll),
  aA_s=accumulate(a.A.begin(),a.A.end(),0ll);
  ll bMA_s=accumulate(b.MA.begin(),b.MA.end(),0ll),
  bA_s=accumulate(b.A.begin(),b.A.end(),0ll);
  for(int i=0;i<CHAR_NUM;i++){
    ret.MMA+=a.A[i]*b.MA[i];
    ret.MMA+=a.A[i]*(a.A[i]-1)/2*(bA_s-b.A[i]);
    ret.MA[i]+=a.A[i]*(bA_s-b.A[i]);
  }
  return ret;
}

int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  int N;
  cin>>N;
  string T;
  cin>>T;
  atcoder::segtree<S,seg_op,seg_e> seg(N);
  for(int i=0;i<N;i++){
    S t=seg_e();
    t.A[T[i]-'A']=1;
    seg.set(i,t);
  }
  int Q;
  cin>>Q;
  while(Q--){
    int a;;
    cin>>a;
    if(a==1){
      int x;
      char c;
      cin>>x>>c;
      --x;
      S t=seg_e();
      t.A[c-'A']=1;
      seg.set(x,t);
    }else if(a==2){
      int l,r;
      cin>>l>>r;
      --l;
      cout<<seg.prod(l,r).MMA<<'\n';
      // seg.prod(l,r).print();
    }
  }
}
0