CF-557 E. Ann and Half-Palindrome(暴力Trie)

題目鏈接

題意

給定一個字符串,求第K個半回文子串.

半回文串:對于字符串$S$, $S_i == S_{n-i+1} $,$i\in[{1, \frac{|s|+1}{2}]}$

思路

暴力找所有合法的區間,首先枚舉長度$i$,

$ok[l][r]=1$表示區間$[l,r]$是一個半回文

$ok[l][r]=(S[l]==R[r])\&\&ok[l+2][r?2]$

找到所有的區間之后在$Trie$上標記每個合法的子串,不能把所有的合法子串插入否則會超時(5000個’$a$’)

之后就$dfs$找第$K$個

#include <bits/stdc++.h> const int maxn = 5e3 + 5; using namespace std; char s[maxn]; bool ok[maxn][maxn]; int k, len; struct Trie{int nex[maxn*maxn/2][2], cnt[maxn*maxn/2], end[maxn*maxn/2], fa[maxn*maxn/2];char C[maxn*maxn/2];int p, root;int newnode() {fill(nex[p], nex[p]+2, 0);cnt[p] = end[p] = 0;return p++;}void init() {p = 0;root = newnode();}void add(int u) {int now = root;for (int i = u; i < len; ++i) {if (nex[now][s[i]-'a'] == 0) {nex[now][s[i] - 'a'] = newnode();C[p-1] = s[i];fa[p-1] = now;}now = nex[now][s[i] - 'a'];if (ok[u][i]) end[now]++;}}void print(int u) {if (fa[u]) print(fa[u]);cout << C[u];}void dfs(int u) {if (end[u]) {k -= end[u];if (k <= 0) {print(u);cout << endl;exit(0);}}for (int i = 0; i < 2; ++i) {if (nex[u][i] == 0) continue;dfs(nex[u][i]);}} }trie; int main() {cin >> s >> k;len = strlen(s);trie.init();for (int i = 1; i <= len; ++i) {for (int j = 0; j <= len-i; ++j) {int l = j, r = j+i-1;if (r - l <= 3) ok[l][r] = (s[l] == s[r]);else ok[l][r] = (s[l] == s[r]) && ok[l+2][r-2];// if (ok[l][r]) trie.add(l, r); 這里太sb了,超時...}}for (int i = 0; i < len; ++i) trie.add(i);trie.dfs(0); return 0; }

總結

以上是生活随笔為你收集整理的CF-557 E. Ann and Half-Palindrome(暴力Trie)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。