Wannafly挑战赛19 F. K串 莫队

链接：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;
}