经验首页 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置:技术经验 » 程序设计 » C++ » 查看文章
BZOJ1396: 识别子串(后缀自动机 线段树)
来源:cnblogs  作者:自为风月马前卒  时间:2018/12/17 9:47:46  对本文有异议

题意

题目链接

Sol

后缀自动机+线段树

还是考虑通过每个前缀的后缀更新答案,首先出现次数只有一次,说明只有\(right\)集合大小为\(1\)的状态能对答案产生影响

设其结束位置为\(t\),代表的最短/最长后缀的位置为\(l, r\)(l在r的右边)

那么对于区间\(r - l\)内的\(x\)位置,可以用\(t - x+1\)更新答案

对于区间\(l - t\)内的位置,可以用\(l\)更新答案

这两种情况不好一起弄(因为第一种情况肯定要把\(x\)提出来),那么直接开两棵线段树就行了

  1. #include<bits/stdc++.h>
  2. #define chmin(a, b) (a = (a < b ? a : b))
  3. #define chmax(a, b) (a = (a > b ? a : b))
  4. using namespace std;
  5. const int MAXN = 1e5 + 10, INF = 1e9 + 10;
  6. char s[MAXN];
  7. int N, fa[MAXN << 1], len[MAXN << 1], ch[MAXN << 1][26], pos[MAXN << 1], siz[MAXN << 1], las = 1, root = 1, tot = 1;
  8. vector<int> v[MAXN << 1];
  9. void insert(int x, int id) {
  10. int now = ++tot, pre = las; las = now; pos[now] = id;
  11. len[now] = len[pre] + 1;
  12. for(; pre && !ch[pre][x]; pre = fa[pre]) ch[pre][x] = now;
  13. if(!pre) fa[now] = root;
  14. else {
  15. int q = ch[pre][x];
  16. if(len[pre] + 1 == len[q]) fa[now] = q;
  17. else {
  18. int ns = ++tot; len[ns] = len[pre] + 1; fa[ns] = fa[q];
  19. memcpy(ch[ns], ch[q], sizeof(ch[q]));
  20. for(; pre && ch[pre][x] == q; pre = fa[pre]) ch[pre][x] = ns;
  21. fa[now] = fa[q] = ns;
  22. }
  23. }
  24. siz[now] = 1;
  25. }
  26. void dfs(int x) {
  27. for(int i = 0; i < v[x].size(); i++) {
  28. int to = v[x][i];
  29. dfs(to); siz[x] += siz[to];
  30. }
  31. }
  32. void Build() {
  33. for(int i = 2; i <= tot; i++) v[fa[i]].push_back(i);
  34. dfs(root);
  35. }
  36. struct SegTree {
  37. #define ls k << 1
  38. #define rs k << 1 | 1
  39. struct Node {
  40. int l, r, mn, f;
  41. Node() {
  42. mn = f = INF;
  43. }
  44. }T[MAXN << 2];
  45. void Build(int k, int ll, int rr) {
  46. T[k].l = ll; T[k].r = rr;
  47. if(ll == rr) return ;
  48. int mid = ll + rr >> 1;
  49. Build(ls, ll, mid); Build(rs, mid + 1, rr);
  50. }
  51. void ps(int k, int val) {
  52. chmin(T[k].f, val);
  53. chmin(T[k].mn, val);
  54. }
  55. void pushdown(int k) {
  56. if(T[k].f == INF) return ;
  57. ps(ls, T[k].f);
  58. ps(rs, T[k].f);
  59. T[k].f = INF;
  60. }
  61. void IntMin(int k, int ll, int rr, int val) {
  62. if(ll <= T[k].l && T[k].r <= rr) {ps(k, val); return ;}
  63. pushdown(k);
  64. int mid = T[k].l + T[k].r >> 1;
  65. if(ll <= mid) IntMin(ls, ll, rr, val);
  66. if(rr > mid) IntMin(rs, ll, rr, val);
  67. }
  68. int PointQuery(int k, int pos) {
  69. if(T[k].l == T[k].r) return T[k].mn;
  70. pushdown(k);
  71. int mid = T[k].l + T[k].r >> 1;
  72. if(pos <= mid) return PointQuery(ls, pos);
  73. if(pos > mid) return PointQuery(rs, pos);
  74. }
  75. }T[2];
  76. int main() {
  77. //freopen("10.in", "r", stdin);freopen("a.out", "w", stdout);
  78. scanf("%s", s + 1); N = strlen(s + 1);
  79. for(int i = 1; i <= N; i++) insert(s[i] - 'a', i);
  80. T[0].Build(1, 1, N); T[1].Build(1, 1, N);
  81. Build();
  82. for(int i = 2; i <= tot; i++) {
  83. if(siz[i] != 1) continue;
  84. int l = pos[i] - len[i] + 1, r = pos[i] - len[fa[i]];
  85. T[0].IntMin(1, l, r, pos[i] + 1);
  86. T[1].IntMin(1, r, pos[i], pos[i] - r + 1);
  87. }
  88. static int mn[MAXN];
  89. memset(mn, 0x3f, sizeof(mn));
  90. for(int i = 1; i <= N; i++) {
  91. chmin(mn[i], T[0].PointQuery(1, i) - i);
  92. chmin(mn[i], T[1].PointQuery(1, i));
  93. }
  94. for(int i = 1; i <= N; i++) printf("%d\n", mn[i]);
  95. return 0;
  96. }
  97. /*
  98. aaa
  99. */
 友情链接:直通硅谷  点职佳  北美留学生论坛

本站QQ群:前端 618073944 | Java 606181507 | Python 626812652 | C/C++ 612253063 | 微信 634508462 | 苹果 692586424 | C#/.net 182808419 | PHP 305140648 | 运维 608723728

W3xue 的所有内容仅供测试,对任何法律问题及风险不承担任何责任。通过使用本站内容随之而来的风险与本站无关。
关于我们  |  意见建议  |  捐助我们  |  报错有奖  |  广告合作、友情链接(目前9元/月)请联系QQ:27243702 沸活量
皖ICP备17017327号-2 皖公网安备34020702000426号