題目描述

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;
}