栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 面试经验 > 面试问答

poj 3376 Finding Palindromes

面试问答 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

poj 3376 Finding Palindromes

#include <cstdio>#include <cstdlib>#include <cstring>#include <algorithm>using namespace std;const int MAX_N = 2000000;char buf[MAX_N + 1], tmp[MAX_N + 1];int n, l[MAX_N], sz[MAX_N], ex[MAX_N], next[MAX_N];char s[MAX_N], t[MAX_N];long long ans = 0LL;namespace trie {    const int MAX_V = MAX_N + 1, MAX_E = 10000000;    int ecnt, begin[MAX_V], to[MAX_E], next[MAX_E], end[MAX_V], cnt[MAX_V];    char val[MAX_V];    int node_idx, root;    void init() {        ecnt = 0;        memset(begin, -1, sizeof(begin));        node_idx = 0;        root = node_idx ++;        val[root] = -1;    }    void add_edge(int u, int v) {        next[ecnt] = begin[u];        begin[u] = ecnt;        to[ecnt ++] = v;    }    void ins(int sz) {        int pos = root;        for (int i = 0; i < sz; i ++) { int t = s[i] - 'a'; if (ex[i] == sz - i) cnt[pos] ++; bool exi = 0; for (int now = begin[pos]; now != -1; now = next[now])     if (val[to[now]] == t) {         exi = 1;         pos = to[now];     } if (!exi) {     int v = node_idx ++;     add_edge(pos, v);     val[v] = t;     pos = v; }        }        end[pos] ++;    }    void go(int sz) {        int pos = root;        for (int i = 0; i < sz; i ++) { int t = s[i] - 'a'; bool exi = 0; for (int now = begin[pos]; now != -1; now = next[now])     if (val[to[now]] == t) {         exi = 1;         pos = to[now];     } if (!exi) return; if (end[pos]) {      if (i < sz - 1 && ex[i + 1] == sz - i - 1) ans += end[pos];      else if (i == sz - 1) ans += end[pos]; }        }        ans += cnt[pos];    }}void ex_kmp(int len) {    next[0] = len;    next[1] = len - 1;    for (int i = 0; i < len - 1; i ++)        if (t[i] != t[i + 1]) { next[1] = i; break;        }    int j, k = 1, p, l;    for (int i = 2; i < len; i ++) {        p = k + next[k] - 1;        l = next[i - k];        if (i + l - 1 < p) next[i] = l;        else { j = max(0, p + 1 - i); while (i + j < len && t[i + j] == t[j]) j ++; next[i] = j, k = i;        }    }    ex[0] = len;    for (int i = 0; i < len; i ++)        if (s[i] != t[i]) { ex[0] = i; break;        }    k = 0;    for (int i = 1; i < len; i ++) {        p = k + ex[k] - 1;        l = next[i - k];        if (i + l - 1 < p) ex[i] = l;        else { j = max(0, p + 1 - i); while (i + j < len && s[i + j] == t[j]) j ++; ex[i] = j, k = i;        }    }}int main() {    scanf("%d", &n);    int tot = 0;    for (int i = 0; i < n; i ++) {        scanf("%d%s", &sz[i], buf + tot);        l[i] = tot;        tot += sz[i];    }    trie::init();    for (int it = 0; it < n; it ++) {        for (int i = 0; i < sz[it]; i ++) s[i] = buf[l[it] + sz[it] - i - 1];        for (int i = 0; i < sz[it]; i ++) t[i] = buf[l[it] + i];        ex_kmp(sz[it]);        trie::ins(sz[it]);        }    for (int it = 0; it < n; it ++) {        for (int i = 0; i < sz[it]; i ++) s[i] = buf[l[it] + i];        for (int i = 0; i < sz[it]; i ++) t[i] = buf[l[it] + sz[it] - i - 1];        ex_kmp(sz[it]);        trie::go(sz[it]);    }    printf("%lldn", ans);    return 0;}
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/366859.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号