Wannafly挑战赛11 D 白兔的字符串【字符串hash】

链接：https://www.nowcoder.com/acm/contest/73/D
来源：牛客网
题目描述
白兔有一个字符串T。白云有若干个字符串S1,S2..Sn。
白兔想知道，对于白云的每一个字符串，它有多少个子串是和T循环同构的。
提示：对于一个字符串a，每次把a的第一个字符移动到最后一个，如果操作若干次后能够得到字符串b，则a和b循环同构。
所有字符都是小写英文字母
输入描述:
第一行一个字符串T（|T|<=10^6）
第二行一个正整数n (n<=1000)
接下来n行为S1~Sn (|S1|+|S2|+…+|Sn|<=10^7)，max(|S1|,|S2|,|S3|,|S4|,..|Sn|)<=10^6
输出描述:
输出n行表示每个串的答案
示例1
输入
复制
abab
2
abababab
ababcbaba
输出
复制
5
2

分析：将字符串hash成一个整数，然后比较就好，map的log级别查找会超时，要手写map。

字符串hash 7道经典问题

#include <bits/stdc++.h>
using namespace std;
typedef unsigned long long ull;

const int mod = 14937;
const int MAXN = 2e6 + 10;
int head[MAXN];
ull base = 233;
ull has[MAXN], a[MAXN];
char str[MAXN];
int cnt = 0, T, len;

struct node {//模拟map
    ull to;
    int next;
}edge[MAXN];

void add_edge(ull x) {
    int pos = x % mod;
    edge[cnt].to = x;
    edge[cnt].next = head[pos];
    head[pos] = cnt++;
}

int find(ull x) {
    int pos = x % mod;
    for(int i = head[pos]; ~i; i = edge[i].next) {
        if(edge[i].to == x) return 1;
    }
    return 0;
}

inline void init_hash() {
    memset(head, -1, sizeof(head));
    a[0] = 1;
    for(int i = 1; i < MAXN; ++i) a[i] = a[i - 1] * base;
    has[0] = 2;
    for(int i = 1; i <= len + len; ++i) {
        has[i] = has[i - 1] * base + str[i];
        if(i >= len) {
            str[i + 1] = str[i - len + 1];
            add_edge(has[i] - has[i - len] * a[len]);
        }
    }
}

int main() {
    scanf("%s", str + 1);
    len = strlen(str + 1);
    init_hash();
    scanf("%d", &T);
    while(T--) {
        scanf("%s", str + 1);
        int tt = strlen(str + 1), ans = 0;
        for(int i = 1; i <= tt; ++i) {
            has[i] = has[i - 1] * base + str[i];
            if(i >= len) ans += find(has[i] - has[i - len] * a[len]);
        }
        printf("%d\n", ans);
    }
    return 0;
}