結果

問題 No.2554 MMA文字列2 (Query Version)
ユーザー Nzt3Nzt3
提出日時 2024-01-12 00:33:10
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 360 ms / 5,000 ms
コード長 3,036 bytes
コンパイル時間 1,954 ms
コンパイル使用メモリ 205,936 KB
実行使用メモリ 57,664 KB
最終ジャッジ日時 2024-09-27 20:41:41
合計ジャッジ時間 16,045 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 3 ms
6,940 KB
testcase_02 AC 3 ms
6,940 KB
testcase_03 AC 4 ms
6,944 KB
testcase_04 AC 5 ms
6,948 KB
testcase_05 AC 5 ms
6,944 KB
testcase_06 AC 3 ms
6,940 KB
testcase_07 AC 3 ms
6,940 KB
testcase_08 AC 3 ms
6,940 KB
testcase_09 AC 5 ms
6,940 KB
testcase_10 AC 5 ms
6,940 KB
testcase_11 AC 2 ms
6,940 KB
testcase_12 AC 41 ms
16,712 KB
testcase_13 AC 282 ms
57,500 KB
testcase_14 AC 57 ms
16,724 KB
testcase_15 AC 257 ms
57,512 KB
testcase_16 AC 155 ms
16,644 KB
testcase_17 AC 237 ms
30,272 KB
testcase_18 AC 80 ms
6,940 KB
testcase_19 AC 112 ms
16,664 KB
testcase_20 AC 281 ms
57,472 KB
testcase_21 AC 98 ms
6,944 KB
testcase_22 AC 327 ms
57,636 KB
testcase_23 AC 330 ms
57,452 KB
testcase_24 AC 335 ms
57,664 KB
testcase_25 AC 330 ms
57,464 KB
testcase_26 AC 337 ms
57,536 KB
testcase_27 AC 335 ms
57,640 KB
testcase_28 AC 352 ms
57,592 KB
testcase_29 AC 350 ms
57,636 KB
testcase_30 AC 339 ms
57,456 KB
testcase_31 AC 336 ms
57,572 KB
testcase_32 AC 31 ms
6,940 KB
testcase_33 AC 31 ms
6,944 KB
testcase_34 AC 36 ms
6,944 KB
testcase_35 AC 36 ms
6,940 KB
testcase_36 AC 37 ms
6,940 KB
testcase_37 AC 327 ms
57,476 KB
testcase_38 AC 323 ms
57,612 KB
testcase_39 AC 316 ms
57,540 KB
testcase_40 AC 323 ms
57,644 KB
testcase_41 AC 323 ms
57,540 KB
testcase_42 AC 326 ms
57,636 KB
testcase_43 AC 318 ms
57,636 KB
testcase_44 AC 350 ms
57,536 KB
testcase_45 AC 323 ms
57,396 KB
testcase_46 AC 325 ms
57,436 KB
testcase_47 AC 346 ms
57,392 KB
testcase_48 AC 348 ms
57,588 KB
testcase_49 AC 358 ms
57,656 KB
testcase_50 AC 348 ms
57,612 KB
testcase_51 AC 353 ms
57,496 KB
testcase_52 AC 351 ms
57,480 KB
testcase_53 AC 346 ms
57,500 KB
testcase_54 AC 360 ms
57,608 KB
testcase_55 AC 348 ms
57,572 KB
testcase_56 AC 348 ms
57,492 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
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;
}

namespace Lib {
template <class S, auto op, auto e>
struct segtree {
  int size, _n;
  vector<S> seg;
  segtree(int n) {
    _n = n;
    size = 1;
    while (size < n) size <<= 1;
    seg = vector<S>(size * 2, e());
  }
  void set(int p, S val) {
    p += size;
    seg[p] = val;
    while (p /= 2) {
      seg[p] = op(seg[p * 2], seg[p * 2 + 1]);
    }
  }
  S prod(int l, int r) {
    S retL = e(), retR = e();
    for (l += size, r += size; l < r; l /= 2, r /= 2) {
      if (l % 2) retL = op(retL, seg[l++]);
      if (r % 2) retR = op(seg[--r], retR);
    }
    return op(retL, retR);
  }
  S get(int p) { return seg[p + size]; }
  template <class F>
  int max_right(int L, F f) {
    L += size;
    S sm = e();
    do {
      while (L % 2 == 0) L /= 2;
      if (!f(op(sm, seg[L]))) {
        while (L < size) {
          L *= 2;
          if (f(op(sm, seg[L]))) {
            sm = op(sm, seg[L]);
            L++;
          }
        }
        return L - size;
      }
      sm = op(sm, seg[L]);
      L++;
    } while ((L & -L) != L);
    return _n;
  }

  template <class F>
  int min_left(int R, F f) {
    R += size;
    S sm = e();
    do {
      R--;
      while (R > 1 && R % 2 == 1) R /= 2;
      if (!f(op(seg[R], sm))) {
        while (R < size) {
          R = R * 2 + 1;
          if (f(op(seg[R], sm))) {
            sm = op(seg[R], sm);
            R--;
          }
        }
        return R + 1 - size;
      }
      sm = op(seg[R], sm);
    } while ((R & -R) != R);
    return 0;
  }
};
}  // namespace Lib
int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  int N;
  cin>>N;
  string T;
  cin>>T;
  Lib::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