Bzoj4598: [Sdoi2016]模式字符串 点分治 哈希

Posted 2020-11-01 Cmd2001

至今不会用Linux的蒟蒻

国际惯例的题面:

这种关于树上路径的题，我也没什么好办法，只好点分治。
考虑当前分治重心为root，如何统计经过分治重心的路径的答案。
我们令prf[i]表示某个点到root的路径(不含root)已经循环匹配S的前缀到位置i(下标从1开始到m-1，结尾为0)的方案数，suf[i]表示某个点到root的路径(不含root)已经循环匹配S的后缀到位置i(下标从1开始到m-1，结尾为0)的方案数。
对于每一个点，考虑当前这个点到root的路径(不含root)加上root作为前缀或者后缀，能和root的其他子树组成的方案数。
那么我们找到当前的匹配长度，读取一下prf和suf在root的其他子树中的前缀和就好了。(像这种维护二元组数量不算自身对自身的有一个套路，就是让自身和前面的和去计算贡献，这样能保证，每组被计算且仅计算一遍)
因为我们要实现在一个串的前面追加一个字符串并查询字符串是否相等，所以要用到哈希(因为这个相等关系不满足单调性)。

然后就是我智障的故事了:
找重心的时候没有重置maxsiz，导致分治被卡n^2且递归深度很大。
一开始在洛谷RE，八中TLE，表示很不服，然后下载了LOJ的数据。
下来测，TLE，发现找分治重心不太对。
然后造了一条链的数据卡他，本地Linux下RE，第一遍重心都找不到。
调系统错误日志发现是崩栈，果断ulimit -s。
然后还是直播崩栈花式崩栈崩栈到怀疑人生，用那个调日志的root终端，发现ulimit -s出来的数值不一样。
又开了一个终端，发现ulimit -s出来的数值还不一样？这才发现ulimit -s仅单个终端有效！
赶紧用我开过栈的终端运行程序，不崩栈了，TLE了。这才发现找重心的问题。

(我可真是菜，这种水题都一遍写不对)

代码:

 1 #pragma GCC optimize(2)
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<algorithm>
 5 #include<cctype>
 6 #include<iostream>
 7 #define debug cerr
 8 using namespace std;
 9 typedef unsigned long long int ulli;
10 using std::max;using std::reverse;
11 const int maxn=1e6+1e2;
12 const ulli base = 31;
13 const int inf=0x3f3f3f3f;
14 
15 char in[maxn],tar[maxn]; // input string and target string .
16 ulli pows[maxn],hpr[maxn],hsu[maxn]; // hash prefix and hash suffix .
17 int s[maxn],t[maxn<<1],nxt[maxn<<1],ban[maxn],cnt;
18 int dep[maxn],siz[maxn],mxs[maxn],prf[maxn],suf[maxn],sprf[maxn],ssuf[maxn]; // mxs[0] = inf  , paired prf[length] .
19 int n,m;
20 ulli ans;
21 
22 bool vis[maxn];
23 
24 inline void addedge(int from,int to) {
25     t[++cnt] = to , nxt[cnt] = s[from] ,s[from] = cnt;
26 }
27 inline void findroot(int pos,const int &fa,const int &fs,int &rt) {
28     siz[pos] = 1 , mxs[pos] = 0;
29     for(int at=s[pos];at;at=nxt[at]) if( t[at] != fa && !ban[t[at]] ) {
30         findroot(t[at],pos,fs,rt) , siz[pos] += siz[t[at]] , mxs[pos] = max( mxs[pos] , siz[t[at]] );
31     }
32     if( ( mxs[pos] = max( mxs[pos] , fs - siz[pos] ) ) <= mxs[rt] ) rt = pos;
33 }
34 inline void dfs(int pos,int fa,int dep,int &mxd,const char &mid,ulli h) {
35     mxd = max( mxd , dep ) , h += in[pos] * pows[dep-1]; // add in[pos] to first char .
36     if( h == hpr[dep] ) {
37         ++prf[dep%m];
38         if( mid == tar[dep%m+1] ) ans += ssuf[m-dep%m-1];
39     }
40     if( h == hsu[dep] ) {
41         ++suf[dep%m];
42         if( mid == tar[m-dep%m] ) ans += sprf[m-dep%m-1];
43     }
44     for(int at=s[pos];at;at=nxt[at]) if( t[at] != fa && !ban[t[at]] ) dfs(t[at],pos,dep+1,mxd,mid,h);
45 }
46 inline void solve(int pos,int fs) {
47     if( fs < m ) return;
48     int rt = 0 , full = 0;
49     *mxs = inf , findroot(pos,-1,fs,rt) , ban[rt] = 1;
50     if( in[rt] == tar[1] ) ++sprf[0];
51     if( in[rt] == tar[m] ) ++ssuf[0];
52     for(int at=s[rt],mxd;at;at=nxt[at]) if( !ban[t[at]] ) {
53         mxd = 0 , dfs(t[at],rt,1,mxd,in[rt],0) , full = max( full , mxd );
54         for(int i=0;i<=mxd;i++) sprf[i] += prf[i] , ssuf[i] += suf[i] , prf[i] = suf[i] = 0;
55     }
56     for(int i=0;i<=full;i++) sprf[i] = ssuf[i] = 0;
57     for(int at=s[rt];at;at=nxt[at]) if( !ban[t[at]] ) solve(t[at],siz[t[at]]<siz[rt]?siz[t[at]]:fs-siz[rt]);
58 }
59 
60 inline char nextchar() {
61     static const int BS = 1 << 21;
62     static char buf[BS],*st=buf+BS,*ed=buf+BS;
63     if( st == ed ) ed = buf + fread(st=buf,1,BS,stdin);
64     return st == ed ? -1 : *st++;
65 }
66 inline int getint() {
67     int ret = 0 , ch;
68     while( !isdigit(ch=nextchar()) );
69     do ret = ret * 10 + ch - \'0\'; while( isdigit(ch=nextchar()) );
70     return ret;
71 }
72 inline void getstr(char* s) {
73     char ch;
74     while( !isalpha(ch=nextchar()) );
75     do *s++ = ch; while( isalpha(ch=nextchar()) );
76 }
77 inline void gethsh(ulli* dst) {
78     for(int i=m+1;i<=n;i++) tar[i] = tar[i-m];
79     for(int i=1;i<=n;i++) dst[i] = dst[i-1] * base + tar[i];
80 }
81 inline void fix(char* s,int len) {
82     for(int i=1;i<=len;i++) s[i] -= \'A\' - 1;
83 }
84 inline void solve_case() {
85     n = getint() , m = getint() , getstr(in+1) , fix(in,n) , memset(s,0,sizeof(s)) , memset(ban,0,sizeof(ban)) , cnt = ans = 0 , *pows = 1;
86     for(int i=1;i<=n;i++) pows[i] = pows[i-1] * base;
87     for(int i=1,a,b;i<n;i++) a = getint() , b = getint() , addedge(a,b) , addedge(b,a);
88     getstr(tar+1) , fix(tar,m) , gethsh(hpr) , reverse(tar+1,tar+1+m) , gethsh(hsu) , reverse(tar+1,tar+1+m);
89     solve(1,n) , printf("%llu\\n",ans);
90 }
91 
92 int main() {
93     static int T;
94     T = getint();
95     while(T--) solve_case();
96     return 0;
97 }

View Code

话说我自带大常数的代码竟然跑到了第一页？


変わらない毎日めぐりめぐる
不变的每日一天又一天
何も無い時間が過ぎる
流逝着无所事事的时间

風にゆらめいて歩き出せば
如果在风中飘舞着迈出步伐
いつか変わるかな？ 明日
明天总会有所改变的吧？

無くしたものを 探しては あきらめて
已逝之物 探寻又放弃
一人佇み 空を見ていた
就这么一个人站着 看着天空