[ural 2118]. Ciphertext
题意
……(语文水平差)
题解
首先如果加密串中同时有串0
和1
,那么答案为-1;如果加密串中同时没有串0
和1
,那么答案为\(|S'|\)。
否则我们假设加密串中只有0
没有1
。
注意到题目给出加密串没有一个是另一个前缀,则一段形如00...01
这样的东西就是合法的。
我们如果找到了一个后缀,满足它不能被解密,那么从它前面的第一个1
后切开,答案就是前面的1
的总数+1。
(即分段的形式为00...01|00...01|...|00...01|UndecodableSuffix
)
显然我们要对于所有不能被解密的后缀求一个最短的。
这个东西只要把反串建AC自动机即可。
复杂度\(\mathcal O(|S|k)\)。
#include <bits/stdc++.h>
using namespace std;
const int N = 1e4 + 5, M = 55;
int n, k, siz;
string oS, S[M];
int c[N * M][2], ed[N * M], fail[N * M];
queue <int> q;
void append (string s) {
int len = s.size(), u = 0;
for (int i = len - 1, x; ~i; --i) {
x = s[i] - '0';
if (!c[u][x]) {
c[u][x] = ++siz;
}
u = c[u][x];
}
ed[u] = 1;
}
void build () {
for (int i = 0; i < 2; ++i) {
if (c[0][i]) {
fail[c[0][i]] = 0;
q.push(c[0][i]);
}
}
for ( ; !q.empty(); q.pop()) {
int u = q.front();
for (int i = 0; i < 2; ++i) {
if (c[u][i]) {
fail[c[u][i]] = c[fail[u]][i];
ed[c[u][i]] |= ed[c[fail[u]][i]];
q.push(c[u][i]);
} else {
c[u][i] = c[fail[u]][i];
}
}
}
}
int id (char c) {
return isupper(c) ? c - 'A' + 1 : c - 'a' + 27;
}
int solve () {
int ret = -1, fl = 0, p = 0, m = oS.size();
for (int i = m - 1, x; i >= 0; --i) {
x = id(oS[i]);
for (int j = S[x].size() - 1; ~j; --j) {
p = c[p][S[x][j] - '0'];
ret += (S[x][j] - '0') * fl;
if (!fl && !ed[p]) {
fl = 1, ret = 1;
}
}
}
return ret;
}
int main () {
cin >> n >> oS, k = 0;
for (int i = 1; i <= n; ++i) {
cin >> S[i];
if (S[i].size() == 1 && S[i] == "0") {
k |= 1;
}
if (S[i].size() == 1 && S[i] == "1") {
k |= 2;
}
}
if (k == 0) {
int ans = 0;
for (int i = 0; i < (int)oS.size(); ++i) {
ans += S[id(oS[i])].size();
}
cout << ans << endl;
return 0;
}
if (k == 3) {
return cout << -1 << endl, 0;
}
if (k == 2) {
for (int i = 1, len; i <= n; ++i) {
len = S[i].size();
for (int j = 0; j < len; ++j) {
S[i][j] = '0' + '1' - S[i][j];
}
}
}
for (int i = 1; i <= n; ++i) {
append(S[i]);
}
build();
cout << solve() << endl;
return 0;
}