Codeforces Round #146 (Div. 1) C - Cyclical Quest (后缀自动机SAM）

最新推荐文章于 2022-11-22 23:15:14 发布
最新推荐文章于 2022-11-22 23:15:14 发布
阅读量440 收藏
点赞数
分类专栏： 后缀自动机
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/kalilili/article/details/78315158
版权

http://codeforces.com/problemset/problem/235/C

陈立杰出的后缀自动机。

题目大意：给一个字符串S，再给一个字符串T,设T的长度为len，问T的循环串在S中出现的次数，这里循环串的定义是：对于一个长度为len的字符串，我们把它首尾相接，然后从任意位置开始走len步所得到的串我们叫做T的循环串。如abaa的循环串有 abaa，baaa，aaab，aaba。（注意如果重复只算一次。比如aaa的循环串只有一个aaa）


思路：对于字符串S，我们构造S的后缀自动机，然后对于每一个字符串T，我们设T'为T去掉最后一个字符所得到的字符串，然后构造TT'，在S的后缀自动机上进行匹配，我们可以算出对于TT'的每一个位置，可以匹配的最大总长度，那么当匹配长度大于等于len时（这里的len为T的长度），设当前所在状态为p，则我们可以根据fa链找到第一个匹配长度大于等于len时所对应的状态，设为q,则我们设状态q所表示的子串出现的次数为q->num，则ans += q->num,num的计算还是通过拓扑排序，自底向上求即可，注意这里有可能有重复，所以我们还得在每一个状态里设一个标记flag，表示当前状态是否被计算过，若已计算过则跳过即可。



数组写法（代码量少，空间小，访问速度快）

//155 ms 
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <iostream>
#include <string>
#include <cmath>
#include <vector>
#include <set>
#include <map>
#include <bitset>
#include <stack>
using namespace std;

#define REP(i,n) for ( int i=1; i<=int(n); i++ )  
#define MP make_pair
#define PB push_back
#define SZ(x) (int((x).size()))
#define ALL(x) (x).begin(), (x).end()
#define X first
#define Y second
template<typename T> inline bool sonkmin(T &a, const T &b) { return a > b ? a = b, 1 : 0; }
template<typename T> inline bool sonkmax(T &a, const T &b) { return a < b ? a = b, 1 : 0; }

typedef long long LL;
typedef long double LD;
const int INF = 0x3f3f3f3f;

const int N = 1e6 + 10;


vector<int> ans;
bool flag[N << 1];
namespace SAM {
  int sz = 0, rt = 0, last = 0;
  int son[N << 1][26], fa[N << 1], val[N << 1], sc[N << 1];
  void init() {
    for(int i = 0; i <= sz; i ++) {
        memset(son[i], 0, sizeof(son[i]));
        fa[i] = val[i] = sc[i] = 0;
    }
    sz = 0; rt = ++ sz; last = rt;
  }
  void add(int c) {
    int p = last, np = ++ sz;
    last = np; val[np] = val[p] + 1;
    sc[np] = 1;
    for (; p && !son[p][c]; p = fa[p]) son[p][c] = np;
    if (p) {
      int q = son[p][c];
      if (val[p] + 1 == val[q]) fa[np] = q;
      else {
        int nq = ++ sz; 
        memcpy(son[nq], son[q], sizeof(son[q]));
        fa[nq] = fa[q], val[nq] = val[p] + 1;
        fa[q] = fa[np] = nq;
        for (; p && son[p][c] == q; p = fa[p]) son[p][c] = nq;
      }
    }
    else fa[np] = rt;
  }
  void getRight(char *s, int n) {
    static int Q[N << 1];
    static int cnt[N];
    for (int i = 0; i <= n; ++ i) cnt[i] = 0;
    for (int p = rt; p <= sz; ++ p) cnt[val[p]] ++;
    for (int i = 1; i <= n; ++ i) cnt[i] += cnt[i - 1];
    for (int p = rt; p <= sz; ++ p) Q[-- cnt[val[p]]] = p;
    for (int i = sz - 1; i >= 0; -- i) {
        int p = Q[i]; if (fa[p]) sc[fa[p]] += sc[p];
    }
  }
  void build(char *s, int n) {
    init();
    for (int i = 0; i < n; ++ i) add(s[i] - 'a');
    getRight(s, n);
  }

  int solve(char *s, int n) {
    int res = 0, len = (n + 1) / 2;
    int p = rt;
    int matson_len = 0;
    for(int i = 0; i < n; i ++) {
        int c = s[i] - 'a';
        while(p && !son[p][c]) p = fa[p];
        if(p) {
            matson_len = min(matson_len, val[p]) + 1;
            p = son[p][c];
        } else p = rt, matson_len = 0;
        if(matson_len >= len) {
            while(val[fa[p]] >= len) {
                p = fa[p];
                matson_len = min(matson_len, val[p]);
            }
            if(flag[p]) continue;
            flag[p] = true;
            ans.push_back(p);
        }
    }
    for(int i = 0; i < ans.size(); i ++)
        flag[ans[i]] = 0, res += sc[ans[i]];
    ans.clear();
    return res;
  }
}

char str[N];
int main() {
    scanf("%s", str);
    SAM::build(str, strlen(str));
    int m;
    scanf("%d", &m);
    while(m --) {
        scanf("%s", str);
        int len = strlen(str);
        for(int i = 0; i < len - 1; i ++) str[len + i] = str[i];
        printf("%d\n", SAM::solve(str, 2 * len - 1));
    }
}


结构体写法（结构清晰，代码量长，访问较慢）

//280 ms
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <iostream>
#include <string>
#include <cmath>
#include <vector>
#include <set>
#include <map>
#include <bitset>
#include <stack>
using namespace std;

#define REP(i,n) for ( int i=1; i<=int(n); i++ )  
#define MP make_pair
#define PB push_back
#define SZ(x) (int((x).size()))
#define ALL(x) (x).begin(), (x).end()
#define X first
#define Y second
template<typename T> inline bool chkmin(T &a, const T &b) { return a > b ? a = b, 1 : 0; }
template<typename T> inline bool chkmax(T &a, const T &b) { return a < b ? a = b, 1 : 0; }

typedef long long LL;
typedef long double LD;
const int INF = 0x3f3f3f3f;

typedef pair<int, int> pii;

const int N = 1e6 + 10;


struct Node {
  int ch[26], fa, val, sc;
  Node(): fa(0), val(0), sc(0) {
    memset(ch, 0, sizeof(ch));
  }
  void clear() {
    memset(ch, 0, sizeof(ch));
    fa = 0; val = sc = 0;
  }
} pool[N << 1];
vector<int> ans;
bool flag[N << 1];
namespace SAM {
  int sz = 0, rt = 0, last = 0;
  void init() {
    for(int i = 0; i <= sz; i ++) pool[i].clear();
    sz = 0; rt = ++ sz; last = rt;
  }
  void add(int c) {
    int p = last, np = ++ sz;
    last = np; pool[np].val = pool[p].val + 1;
    pool[np].sc = 1;
    for (; p && !pool[p].ch[c]; p = pool[p].fa) pool[p].ch[c] = np;
    if (p) {
      int q = pool[p].ch[c];
      if (pool[p].val + 1 == pool[q].val) pool[np].fa = q;
      else {
        int nq = ++ sz; 
        memcpy(&pool[nq], &pool[q], sizeof(pool[q]));
        pool[nq].val = pool[p].val + 1, pool[nq].sc = 0;
        pool[q].fa = nq; pool[np].fa = nq;
        for (; p && pool[p].ch[c] == q; p = pool[p].fa) pool[p].ch[c] = nq;
      }
    }
    else pool[np].fa = rt;
  }
  void getRight(char *s, int n) {
    static int Q[N << 1];
    static int cnt[N];
    for (int i = 0; i <= n; ++ i) cnt[i] = 0;
    for (int p = rt; p <= sz; ++ p) cnt[pool[p].val] ++;
    for (int i = 1; i <= n; ++ i) cnt[i] += cnt[i - 1];
    for (int p = rt; p <= sz; ++ p) Q[-- cnt[pool[p].val]] = p;
    for (int i = sz - 1; i >= 0; -- i) {
        int p = Q[i]; if (pool[p].fa) pool[pool[p].fa].sc += pool[p].sc;
    }
  }
  void build(char *s, int n) {
    init();
    for (int i = 0; i < n; ++ i) add(s[i] - 'a');
    getRight(s, n);
  }

  int solve(char *s, int n) {
    int res = 0, len = (n + 1) / 2;
    int p = rt;
    int match_len = 0;
    for(int i = 0; i < n; i ++) {
        int c = s[i] - 'a';
        while(p && !pool[p].ch[c]) p = pool[p].fa;
        if(p) {
            match_len = min(match_len, pool[p].val) + 1;
            p = pool[p].ch[c];
        } else p = rt, match_len = 0;
        if(match_len >= len) {
            while(pool[pool[p].fa].val >= len) {
                p = pool[p].fa;
                match_len = min(match_len, pool[p].val);
            }
            if(flag[p]) continue;
            flag[p] = true;
            ans.push_back(p);
        }
    }
    for(int i = 0; i < ans.size(); i ++)
        flag[ans[i]] = 0, res += pool[ans[i]].sc;
    ans.clear();
    return res;
  }
}

char str[N];
int main() {
    scanf("%s", str);
    SAM::build(str, strlen(str));
    int m;
    scanf("%d", &m);
    while(m --) {
        scanf("%s", str);
        int len = strlen(str);
        for(int i = 0; i < len - 1; i ++) str[len + i] = str[i];
        printf("%d\n", SAM::solve(str, 2 * len - 1));
    }
}

结构体指针写法（结构清晰，代码量小，访问慢，指针空间消耗大） 


//280 ms
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <iostream>
#include <string>
#include <cmath>
#include <vector>
#include <set>
#include <map>
#include <bitset>
#include <stack>
using namespace std;

#define REP(i,n) for ( int i=1; i<=int(n); i++ )  
#define MP make_pair
#define PB push_back
#define SZ(x) (int((x).size()))
#define ALL(x) (x).begin(), (x).end()
#define X first
#define Y second
template<typename T> inline bool chkmin(T &a, const T &b) { return a > b ? a = b, 1 : 0; }
template<typename T> inline bool chkmax(T &a, const T &b) { return a < b ? a = b, 1 : 0; }

typedef long long LL;
typedef long double LD;
const int INF = 0x3f3f3f3f;

typedef pair<int, int> pii;

const int N = 1e6 + 10;


struct Node {
  Node *ch[26], *fa;
  int val, sc;
  Node(): fa(0), val(0), sc(0) {
    memset(ch, 0, sizeof(ch));
  }
  void clear() {
    memset(ch, 0, sizeof(ch));
    fa = 0; val = sc = 0;
  }
} pool[N << 1], *rt, *last;
vector<Node *> ans;
bool flag[N << 1];
namespace SAM {
  Node *sz = pool;
  void init() {
    if (sz != pool) {
      for (Node *p = pool; p < sz; ++ p) p->clear();
    }
    sz = pool; rt = sz ++; last = rt;
  }
  void add(int c) {
    Node *p = last, *np = sz ++;
    last = np; np->val = p->val + 1;
    np->sc = 1;
    for (; p && !p->ch[c]; p = p->fa) p->ch[c] = np;
    if (p) {
      Node *q = p->ch[c];
      if (p->val + 1 == q->val) np->fa = q;
      else {
        Node *nq = sz ++; *nq = *q;
        nq->sc = 0;
        nq->val = p->val + 1;
        q->fa = nq; np->fa = nq;
        for (; p && p->ch[c] == q; p = p->fa) p->ch[c] = nq;
      }
    }
    else np->fa = rt;
  }
  void getRight(char *s, int n) {
    static Node* Q[N << 1];
    static int cnt[N];
    for (int i = 0; i <= n; ++ i) cnt[i] = 0;
    for (Node *p = pool; p < sz; ++ p) cnt[p->val] ++;
    for (int i = 1; i <= n; ++ i) cnt[i] += cnt[i - 1];
    for (Node *p = pool; p < sz; ++ p) Q[-- cnt[p->val]] = p;
    for (int i = (sz - pool) - 1; i >= 0; -- i) {
      Node *p = Q[i]; if (p->fa) p->fa->sc += p->sc;
    }
  }
  void build(char *s, int n) {
    init();
    for (int i = 0; i < n; ++ i) add(s[i] - 'a');
    getRight(s, n);
  }

  int solve(char *s, int n) {
    int res = 0, len = (n + 1) / 2;
    Node *p = rt;
    int match_len = 0;
    for(int i = 0; i < n; i ++) {
        int c = s[i] - 'a';
        while(p && !p->ch[c]) p = p->fa;
        if(p) {
            match_len = min(match_len, p->val) + 1;
            p = p->ch[c];
        } else p = rt, match_len = 0;
        if(match_len >= len) {
            while(p != rt && p->fa->val >= len) {
                p = p->fa;
                match_len = min(match_len, p->val);
            }
            if(flag[p - pool]) continue;
            flag[p - pool] = true;
            ans.push_back(p);
        }
    }
    for(int i = 0; i < ans.size(); i ++)
        flag[ans[i] - pool] = 0, res += ans[i]->sc;
    ans.clear();
    return res;
  }
}

char str[N];
int main() {
    scanf("%s", str);
    SAM::build(str, strlen(str));
    int m;
    scanf("%d", &m);
    while(m --) {
        scanf("%s", str);
        int len = strlen(str);
        for(int i = 0; i < len - 1; i ++) str[len + i] = str[i];
        printf("%d\n", SAM::solve(str, 2 * len - 1));
    }
}


kalilili
关注
专栏目录
Codeforces Round #806 (Div. 4)参考代码
guajiProMax的博客
07-13 470
Codeforces Round #806 (Div. 4)参考代码
Codeforces Round #787 (Div. 3)个人题解
weixin_51767904的博客
05-06 1098
Codeforces Round #787 (Div. 3)个人题解 Codeforces Round #787 (Div. 3) A. Food for Animals B. Make It Increasing C. Detective Task D. Vertical Paths E. Replace With the Previous, Minimize F. Vlad and Unfinished Business G. Sorting Pancakes
Codeforces Round #146 (Div. 1) C - Cyclical Quest 后缀自动机+最小循环节
10-21 125
#include<bits/stdc++.h> #define LL long long #define fi first #define se second #define mk make_pair #define PII pair<int, int> #define PLI pair<LL, int> #define ull un...
Codeforces Round #146 (Div. 1) B. Let's Play Osu!（期望DP）
qq_42479630的博客
03-12 184
题目链接 #include<bits/stdc++.h> #define N 100005 using namespace std; int n; double a, f1, f2; int main () { scanf ("%d", &n); for (int i = 1; i <= n; ++i) { scanf ("%lf", ...
Codeforces Round #146 (Div. 1) C. Cyclical Quest——后缀自动机
cool_Fires的专栏
08-04 865
题意：给定一个母串，每次询问一个子串或把子串的前一部分移到末尾形成的串在母串中出现的次数。 处理方法都是很常见的。对询问中的每个子串，设长度为len。复制一份放到其末尾位置并把最后一个字符去掉。再拿这个串和母串的自动机匹配，对于询问串的每一个位置，记录该位置往前能匹配的长度，如果长度大于等于len，则通过pre指针找到从这个位置往前len个字符之间的串在自动机中对应的状态。并将答案加上该状态出现
Codeforces Round #146 (Div. 1) LCM Challenge 解题报告 数论 找规律
学无止境
07-16 1785
题目链接：http://codeforces.com/problemset/problem/235/A 题目意思：找出三个小于 n 的数，这三个数组成的 LCM（最小公倍数）是最大的。 首先，这三个数应该是互质的。根据 LCM 的公式 (a * b) / gcd(a,b)，gcd(a,b)为1时，即互质，LCM 值是最大的。 然后开始找规律： 如果是奇数的话（n=7），那么
Codeforces Round #146 (Div. 2) B. Easy Number Challenge
shiyuan的专栏
10-23 1142
B. Easy Number Challenge time limit per test 2 seconds memory limit per test 256 megabytes input standard input output standard output Let's denote d(n) as the number of
Codeforces Round #723 (Div. 2).md
10-13
Codeforces Round #723 (Div. 2).md
Codeforces Round #680 (Div. 2, based on Moscow Team Olympiad)
qq_45074215的博客
11-01 335
C #include<iostream> #include<cstring> #include<algorithm> #include<set> #include<map> #include<unordered_map> #include<cmath> #include<unordered_set> #include<queue> using namespace std; #define int lo
Codeforces Round #833 (Div. 2) 思路
m0_75212691的博客
11-22 280
Codeforces Round #833 (Div. 2) 思路
Codeforces Round #146 (Div. 1) E. Number Challenge 莫比乌斯反演
kaka03200的博客
03-29 180
原题链接：https://codeforces.ml/contest/235/problem/E 目录题意化简Code 题意 ∑i=1a∑j=1b∑k=1cd(ijk)\sum_{i=1}^{a}\sum_{j=1}^{b}\sum_{k=1}^{c}d(ijk)i=1∑a​j=1∑b​k=1∑c​d(ijk) d(n)代表n的因子个数d(n)代表n的因子个数d(n)代表n的因子个数 化简 已知结论d(ij)=∑x∣i∑y∣j[gcd(x,y)=1]已知结论d(ij)=\sum_{x|i}\sum_{y|j
Codeforces Round #146 E. Number Challenge 莫比乌斯反演
xgc_woker的博客
12-29 162
Description 定义d(i)d(i)d(i)为iii的约数个数，求： ∑i=1a∑j=1b∑k=1cd(ijk)\sum_{i=1}^a\sum_{j=1}^b\sum_{k=1}^cd(ijk)i=1∑a​j=1∑b​k=1∑c​d(ijk) Sample Input 2 2 2 Sample Output 20 首先有一个性质： ∑i=1a∑j=1bd(ij)=∑i=1a∑j=1...
解题报告：Codeforces Round #146 (Div. 1) B. Let's Play Osu! 概率DP
Hee的博客~
07-16 307
题目链接 题意： 有n个位置，每个 位置 i 为圈的概率给出，一段序列的值为每段连续圈的长度的平方和，求序列的值的期望 思路： 若我们已经知道了长度为(x-1)的期望F(x-1) ，我们要求长度为x的期望F(x) 那么假设最后一位不为圈，那么此时的贡献为：F(x-1)  *  ( 1 - p ) 若最后一位为圈，那么此时的贡献为以它结尾的每段连续圈的贡献和，写成式子如下
Problem-236A-Codeforces-Boy or Girl
qq_36667106的博客
03-22 1753
A. Boy or Girl time limit per test1 second memory limit per test256 megabytes inputstandard input outputstandard output Those days, many boys use beautiful girls’ photos as avatar
解题报告：Codeforces Round #146 (Div. 1) E. Number Challenge 莫比乌斯反演
Hee的博客~
08-12 519
题目链接 题意： 给定n,m,l，求 d(x)：x的约数个数 思路： 加强版的约数个数和 （ 解题报告：BZOJ_3994 约数个数和 莫比乌斯反演学习题）？ 代入公式 得到 得到 代入常用的公式： 得到： 这个式子的复杂度为 但是可以发现对于一个数k，只需要用到和它互质的部分 那么可以在的时间内预测出每个数的互质的所
Codeforces200E解题报告
TA
05-04 1363
这道题虽然是水题，但是我还是犯了很多傻逼错误，导致比赛的时候挂了，之后又对着数据才调出来。。 犯的错误有： ①把floor写成了ceil。。floor是下取整，floor(-0.5)=-1;ceil是上取整。 ②对拍的时候忘了srand…导致一直在拍一组数据。 ③|x−a|+|x−b|(a≤b)|x-a|+|x-b|(a \le b)应该在[a,b]处取最大值
ACM牛人集
Geecky的博客
06-21 879
楼天成——楼教主 陈立杰——高二学生，codeforces全球排名第五，入选国家OI集训队，发明后缀自动机。可能是因为codeforces的排名，高三意外得到google总部的offer。
注册了codeforces账号，ACM选手的代码真滴漂亮！
Sunshine_xiaohao的博客
11-27 815
如题！ 看了一道div2的A题，点开了top1的提交记录，不仅这道题的想法震撼了我，他的预编译部分让我学到了好多嗷，先睡个午觉，有空更。
codeforces round #792 (div. 1 + div. 2)
最新发布
06-06
codeforces round #792 (div. 1 + div. 2) 是 Codeforces 竞赛平台上的一场比赛，包括两个级别：div. 1 和 div. 2。这场比赛通常会有多个问题需要参赛者解决，每个问题都有不同的难度等级。参赛者需要在规定的时间内...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值