【ACWing】3723. 字符串查询

题目地址：

https://www.acwing.com/problem/content/description/3726/

给你单词$S$和$Q$个询问。每次询问，你会得到正整数$A,B,C$和$D$。我们令单词$X$由$S$的第$A$到$B$个字母组成，单词$Y$由$S$的第$C$到$D$个字母组成。你需要回答，是否能够重新排列单词$Y$中的字母，得到单词$X$。

输入格式：
第一行一个单词$S$，仅由小写字母组成。
第二行一个正整数$Q$。
接下来$Q$行，每行四个整数$A,B,C,D$。

输出格式：
每次询问，如果能，输出DA，否则输出NE。

数据范围：
$1\le |S|\le 50000$,
$1\le Q\le 50000$,
$1\le A\le B\le |S|$,
$1\le C\le D\le |S|$,
$B-A=D-C$

可以用前缀和来处理区间里每个字母出现的次数，询问的时候比较每个字母出现次数是否相等即可。代码如下：

#include <cstring>
#include <iostream>
using namespace std;

const int N = 5e4 + 10;
char s[N];
int n, q;
int a, b, c, d;
int cnt[26][N];

int main() {
  scanf("%s", s + 1);
  n = strlen(s + 1);
  for (int i = 1; i <= n; i++) {
    for (int j = 0; j < 26; j++) cnt[j][i] = cnt[j][i - 1];
    cnt[s[i] - 'a'][i]++;
  }

  scanf("%d", &q);
  while (q--) {
    scanf("%d%d%d%d", &a, &b, &c, &d);
    bool res = true;
    for (int i = 0; i < 26; i++)
      if (cnt[i][b] - cnt[i][a - 1] != cnt[i][d] - cnt[i][c - 1]) {
        res = false;
        break;
      }
    res ? puts("DA") : puts("NE");
  }
}

预处理时间$O(l_S)$，空间$O(l_S)$，每组询问时间复杂度$O(1)$。