嗨呀,用了std::ios::sync_with_stdio(false),比答案块一丢丢。
用了一点DP的思想。对每个字符,因为prefix最长为十的原因,只需要,检测其前缀(从长度为一到长度为十,包括该字符本身)是否在primitive集合中。
/*
ID :
LANG: C++11
TASK: prefix
*/
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <string>
#include <set>
using namespace std;
set<string> primitives;
int main()
{
freopen("prefix.in", "r", stdin);
freopen("prefix.out", "w", stdout);
std::ios::sync_with_stdio(false);
int ans = 0;
int p = -1;
string s, tail; //每次读入的字符串以及,上一行字符串的结尾10个字符,这十个字符可能和下一行的前几位相连且在primitives集合里
while (cin >> s){
if (s == ".")
break;
primitives.insert(s);
}
bool bOK[15] = {}; //记录,当前字符,往前的10个位置,是否是一个prefix
bool bEnd = false; //结束符号,如果连续十个位置都不是prefix,则结束
while (cin >> s){ //读入测试数据
s = tail + s;
int i = tail.length();
p = -1; //当前最长prefix的位置
for (int i = 0; i <= 9; i ++){ //计算上一行字符串的tail连到当前行时,最长prefix的位置
if (bOK[i])
p = i;
}
while (s[i] != 0){
for (int j = 1; j <= 10; j ++){ //当前位置i,如果从开头到i-j是一个prefix,且i-j到i在primitives集合中,则从开头到i也是一个prefix
if (i - j < 0){ //处理开头前十个
if (primitives.find(s.substr(0, j)) != primitives.end()){
bOK[10] = true;
}
break;
}
else{
if (bOK[10 - j] && (primitives.find(s.substr(i - j + 1 , j)) != primitives.end())){
bOK[10] = true;
break;
}
}
}
if (bOK[10]){
ans += i - p;
p = i;
}
bool count = false;
for (int i = 1; i <= 10; i ++){
if (bOK[i]){
count = true;
}
bOK[i - 1] = bOK[i];
}
bOK[10] = false;
if (!count && i >= 9){
bEnd = true;
break;
}
i ++;
}
if (bEnd){
break;
}
tail = i >= 10 ? s.substr(i - 10, 10) : s;
}
cout << ans << endl;
return 0;
}