結果

問題 No.1725 [Cherry 3rd Tune D] 無言の言葉
ユーザー stoq
提出日時 2021-07-01 05:57:52
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 93 ms / 4,000 ms
コード長 1,682 bytes
コンパイル時間 5,756 ms
コンパイル使用メモリ 199,352 KB
最終ジャッジ日時 2025-01-22 15:13:15
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
#define rep(i, n) for (int i = 0; i < int(n); i++)
#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")

int n, m;                             // X, Y の長さ
int len[32];                          // F_k の長さ
int F_all[32][26];                    // F_k 全体に含まれる c の個数
int X_L[100010][26], Y_L[100010][26]; // X, Y の先頭 i 文字に含まれる c の個数

// F_k の [0, r) に含まれる c の個数
int f(int k, int r, int c)
{
  if (k == 0)
    return X_L[r][c];
  assert(r <= len[k]);
  // F_k = F_{k-1} + Y + rev(F_{k-1})
  if (r <= len[k - 1])
    return f(k - 1, r, c);
  else if (r <= len[k - 1] + m)
    return F_all[k - 1][c] + Y_L[r - len[k - 1]][c];
  else
    return F_all[k - 1][c] + Y_L[m][c] + F_all[k - 1][c] - f(k - 1, len[k] - r, c);
}

int main()
{
  cin.tie(0);
  ios::sync_with_stdio(false);

  string X, Y;
  int Q;
  cin >> X >> Y >> Q;
  n = X.length(), m = Y.length();
  rep(j, 26) X_L[0][j] = Y_L[0][j] = 0;
  rep(i, n) rep(j, 26)
  {
    X_L[i + 1][j] += X_L[i][j] + (X[i] == j + 'a');
  }
  rep(i, m) rep(j, 26)
  {
    Y_L[i + 1][j] += Y_L[i][j] + (Y[i] == j + 'a');
  }

  int k = -1;
  rep(i, 32)
  {
    if (i == 0)
    {
      len[i] = n;
      rep(j, 26) F_all[0][j] = X_L[n][j];
    }
    else
    {
      len[i] = len[i - 1] * 2 + m;
      rep(j, 26) F_all[i][j] = F_all[i - 1][j] * 2 + Y_L[m][j];
    }
    if (len[i] >= 1000000010)
    {
      k = i;
      break;
    }
  }
  assert(k != -1);

  rep(i, Q)
  {
    int l, r;
    char c;
    cin >> l >> r >> c;
    cout << f(k, r, c - 'a') - f(k, l - 1, c - 'a') << "\n";
  }
}
0