链接:https://www.nowcoder.com/acm/contest/131/F
来源:牛客网
题目描述
ZZT 得到了一个字符串 S 以及一个整数 K。
WZH 在 1995 年提出了“优雅 K 串”的定义:这个字符串每一种字符的个数都是 K 的倍数。
现在 ZZT 想要对字符串进行 Q 次询问,第 i 次询问给出一个区间 [Li, Ri],他想计算 [Li, Ri] 中有多少个子串是“优雅 K 串”。
由于 ZZT 忙于工作,所以他把这个问题交给了你,请你帮忙解决。
输入描述:
第一行输入一个正整数 K。 第二行输入一个字符串 S。 第三行输入一个正整数 Q,表示有 Q 次询问。
接下来 Q 行,每行输入两个正整数 Li 和 Ri,表示第 i 次询问。
1 ≤ K ≤ 50.
1≤ | S | ≤ 3 x 104 且 S 仅包含小写英文字母.
1≤ Q ≤ 3 x 104.
1 ≤ Xi ≤ Yi ≤ N.
输出描述:
每次询问,输出一个正整数,表示满足条件的“优雅 K 串”的数量。
题解:
莫队即可。
代码:
#include <bits/stdc++.h>
#ifdef LOCAL
#define debug(x) cout<<#x<<" = "<<(x)<<endl;
#else
#define debug(x) 1;
#endif
#define chmax(x,y) x=max(x,y)
#define chmin(x,y) x=min(x,y)
#define lson id<<1,l,mid
#define rson id<<1|1,mid+1,r
#define lowbit(x) x&-x
#define mp make_pair
#define pb push_back
#define fir first
#define sec second
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
const int MOD = 1e9 + 7;
const double PI = acos (-1.);
const double eps = 1e-10;
const int INF = 0x3f3f3f3f;
const ll INFLL = 0x3f3f3f3f3f3f3f3f;
const int MAXN = 5e5 + 5;
int k, q, n;
char s[MAXN];
int block;
ull val[MAXN];
int pre[MAXN][26];
unordered_map <ull, int> vis;
struct Query {
int l, r, id;
bool operator < (const Query & x) const {
return l / block < x.l / block || (l / block == x.l / bloc);
}
} qr[MAXN];
ll res[MAXN];
int main() {
#ifdef LOCAL
freopen ("input.txt", "r", stdin);
#endif
scanf ("%d %s %d", &k, s + 1, &q);
for (int i = 1; i <= q; i++) scanf ("%d %d", &qr[i].l, &qr[i].r), qr[i].id = i;
n = strlen(s + 1);
for (int i = 1; i <= n; i++) {
for (int j = 0; j < 26; j++) {
pre[i][j] = (pre[i - 1][j] + (s[i] - 'a' == j)) % k;
val[i] = val[i] * 233 + pre[i][j];
}
}
block = sqrt (n + 0.5);
sort (qr + 1, qr + 1 + q);
int l = 1, r = 0;
ll ans = 0;
for (int i = 1; i <= q; i++) {
qr[i].l--;
while (r < qr[i].r) {
r++;
ans += vis[val[r]];
vis[val[r]]++;
}
while (r > qr[i].r) {
vis[val[r]]--;
ans -= vis[val[r]];
r--;
}
while (l < qr[i].l) {
vis[val[l]]--;
ans -= vis[val[l]];
l++;
}
while (l > qr[i].l) {
l--;
ans += vis[val[l]];
vis[val[l]]++;
}
res[qr[i].id] = ans;
}
for (int i = 1; i <= q; i++) printf ("%lld\n", res[i]);
return 0;
}