java实现 蓝桥杯 算法训练 Password Suspects
问题描述
在年轻的时候，我们故事中的英雄——国王 Copa——他的私人数据并不是完全安全地隐蔽。对他来说是，这不可接受的。因此，他发明了一种密码，好记又难以破解。后来，他才知道这种密码是一个长度为奇数的回文串。
Copa 害怕忘记密码，所以他决定把密码写在一张纸上。他发现这样保存密码不安全，于是他决定按下述方法加密密码：他选定一个整数 X ，保证 X 不小于 0 ，且 2X 严格小于串长度。然后他把密码分成 3 段，最前面的 X 个字符为一段，最后面的 X 个字符为一段，剩余的字符为一段。不妨把这三段依次称之为 prefix, suffix, middle 。显然， middle 的长度为一个大于 0 的奇数，且 prefix 、 suffix 的长度相等。他加密后的密码即为 A + prefix + B + middle + C + suffix ，其中 A 、 B 、 C 是三个由 Copa 选定的字符串，且都有可能为空， + 表示字符串相连。
许多年过去了。Copa 昨天找到了当年写下加密后字符串的那张纸。但是，Copa 把原密码、A、B、C 都忘了。现在，他请你找一个尽量长的密码，使得这个密码有可能被当年的 Copa 发明、加密并写下。
输入格式
输入包含一个只含有小写拉丁字母的字符串，长度在 1 到 10^5 之内。
输出格式
第一行包含一个整数 k ，表示你找到的原密码分成的 3 个部分中有多少个非空字符串。显然 k in {1, 3} 。接下来 k 行，每行 2 个用空格分开的整数 x_i l_i ，表示这一部分的起始位置和长度。要求输出的 x_i 递增。
起始位置 x_i 应该在 1 到加密后的字符串长度之间。 l_i 必须是正整数，因为你只要输出非空部分的信息。 middle 的长度必须为奇数。
如果有多组答案，任意一组即可。提示：你要最大化的是输出的 l_i 的总和，而不是 k 。
样例输入
abacaba
样例输出
1
1 7
样例输入
axbya
样例输出
3
1 1
2 1
5 1
样例输入
xabyczba
样例输出
3
2 2
4 1
7 2
数据规模和约定
对于 10% 的数据： n <= 10
对于 30% 的数据： n <= 100
对于 100% 的数据： n <= 100000
存在 20% 的数据，输出文件第一行为 1 。
刚开始四十分的思路（下面有AC代码）
输入一段字符串组成为： A + prefix + B + middle + C + suffix 。现在要求取 prefix + middle + suffix ，由于题目说：他选定一个整数 X ，保证 X 不小于 0 ，且 2X 严格小于串长度。那么X可取0。即当X = 0时，也只有一段——middle。
现在要求取最长的密码串：
（1）首先求取原字符串中的最长回文串长度，用maxLen0表示。
（2）由于suffix在字符串尾部，先将suffix反转，然后使用KMP算法，在原字符串中找到第一次与suffix匹配的字符串，记录suffix长度文len1。然后，求取者两端匹配字符串中间剩余的一段字符串的最大回文串的长度len2。此时，求取的密码长度maxLen = 2*len1 + len2。
在这种情况下，影响maxLen长度的因素就是初始选择的len1长度，len1长度变化，也会导致len2长度的变化。所以，我这里采用1 <= len1 < 原始字符串一半的方法来枚举，求取其中的最大maxLen。个人感觉就是在这里，才导致运算超时…
package com.liuzhen.systemExe;
import java.util.ArrayList;
import java.util.Scanner;
public class Main{
//找出字符串arrayA[i]到arrayA[j]中的最大回文串，并返回其初始位置和长度（PS：此方法称作中心扩展法）
public int[] getMaxReverse(char[] arrayA, int i, int j) {
int[] result = new int[2];
int max = 1;
int begin = i+1; //最大回文串的开始位置，初始化为字符i的位置
int zero = i; //保存i的初始值
for(i = i + 1;i < j;i++) {
int start = i - 1;
int end = i + 1;
int count = 1;
while(start >= zero && end <= j) { //对于此种情况，只返回奇数位回文串
if(arrayA[start] == arrayA[end]) {
start--;
end++;
count = count + 2;
}
else
break;
}
if(count > max) {
max = count;
begin = i - (count-1)/2 + 1;
}
}
result[0] = begin; //最大回文串开始位置
result[1] = max; //最大回文串长度
return result;
}
//此处代码是使用Manacher算法求取最大回文串，但是在蓝桥杯练习系统中评分时，也是运行超时，所以在求最大回文串就采用上面更易理解的方法
// public int[] getManacher(char[] arrayA, int i, int j) {
// int[] result = new int[2];
// ArrayList listA = new ArrayList();
// for(int y = i;y <= j;y++) {
// listA.add('#');
// listA.add(arrayA[y]);
// }
// listA.add('#');
// int[] P = new int[listA.size() + 1];
// int mx = 0;
// int d = 0;
// for(int y = 1;y <= listA.size();y++) {
// if(mx > y) {
// P[y] = (P[2 * d - y] > (mx - y) ? (mx - y) : P[2 * d - y]);
// } else {
// P[y] = 1;
// }
//
// while(y + P[y] < listA.size() && y - P[y] >= 0 && listA.get(y + P[y]) == listA.get(y - P[y]))
// P[y]++;
//
// if(mx < y + P[y]) {
// mx = y + P[y];
// d = y;
// }
// }
//
// int max = 0;
// int tempY = 0;
// for(int y = 0;y < P.length;y++) {
// if(P[y] > max) {
// max = P[y];
// tempY = y;
// }
// }
// int begin = (tempY - 1)/2;
// begin = begin - (max - 1) / 2 + i + 1;
// result[0] = begin;
// result[1] = max - 1;
// return result;
// }
//使用KMP模式匹配，计算next函数值
public int[] getNext(char[] arrayB) {
int[] next = new int[arrayB.length + 1];
int j = 0;
for(int i = 1;i < arrayB.length;i++) {
while(j > 0 && arrayB[i] != arrayB[j]) j = next[j];
if(arrayB[i] == arrayB[j]) j++;
next[i+1] = j;
}
return next;
}
//使用KMP算法，找出字符数组arrayB，在arrayA中第一次出现匹配的位置
public int getKmpFirstPosition(char[] arrayA, char[] arrayB) {
int position = -1;
int j = 0;
int[] next = getNext(arrayB);
for(int i = 0;i < arrayA.length;i++) {
while(j > 0 && arrayA[i] != arrayB[j]) j = next[j];
if(arrayA[i] == arrayB[j])
j++;
if(j == arrayB.length) {
position = i - j + 1;
break;
}
}
return position;
}
public void printResult(String A) {
//当选定整数X = 0时，输出第一行为1，此时只需在A中直接找到有段最长回文串即可，这时的密码最长
char[] arrayA = A.toCharArray();
int[] result0 = getMaxReverse(arrayA, 0, arrayA.length-1);
int maxLen0 = result0[1]; //获得此时回文串的最大长度
//当X ！= 0时，最长密码其尾部一定在输入的字符串尾部，即 suffix不为空
int j = arrayA.length - 1;
int maxLen = 0; //用于计算最长密码，初始化为0
int position1 = 0;
int position2 = 0;
int position3 = 0;
int len1 = 0;
int len2 = 0;
for(int lenS = 1;lenS < arrayA.length/2;lenS++) {
char[] tempS = new char[lenS];
for(int a = 0;a < lenS;a++)
tempS[a] = arrayA[j-a];
if(getKmpFirstPosition(arrayA, tempS) == -1) {
continue;
}
int tempPosition1 = getKmpFirstPosition(arrayA, tempS) + 1;
int endPosition1 = tempPosition1+lenS;
int startPosition3 = arrayA.length-lenS;
if(startPosition3 <= endPosition1)
continue;
int[] result = getMaxReverse(arrayA,tempPosition1+lenS-1,j-lenS);
int tempLen2 = result[1];
int tempPosition2 = result[0];
if(lenS * 2 + tempLen2 > maxLen) {
position1 = tempPosition1;
position2 = tempPosition2;
position3 = j - lenS + 2;
len1 = lenS;
len2 = tempLen2;
maxLen = lenS * 2 + tempLen2;
}
}
if(maxLen0 >= maxLen) {
System.out.println("1");
System.out.println(result0[0]+" "+result0[1]);
} else {
System.out.println("3");
System.out.println(position1+" "+len1);
System.out.println(position2+" "+len2);
System.out.println(position3+" "+len1);
}
}
public static void main(String[] args){
Main test = new Main();
Scanner in = new Scanner(System.in);
// System.out.println("请输入一个字符串：");
String A = in.nextLine();
test.printResult(A);
}
}
大概思路
d[u][len][state]表示当前递归到了这样一个状态：
1.当前密码串已经确定的长度是len
2.用字典树对已确定的密码串进行匹配，到当前状态时，字典树匹配到的结点是u
3.如果state的二进制数是00001010B，假设密码子串有5个，则表示当前状态已经出现的密码子串有按输入顺序的第2、4个。
这里实际是一个状态压缩，将所有密码的可能压缩到了三个参数上。
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.concurrent.ConcurrentLinkedQueue;
/**
*
* 这个题不是原题，原题来自UVALive
* 方法是使用AC自动机（要理解AC自动机需要先掌握KMP算法和字典树） 以及DP算法
* 参考实现CSDN
*/
public class Main {
public static void main(String[] args) {
int n = 0;
int m = 0;
try {
// 读取数据
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
String[] nm = br.readLine().split(" ");
n = Integer.parseInt(nm[0]);
m = Integer.parseInt(nm[1]);
// 构建AC自动机
AC ac = new AC(n, m);
for(int i = 0; i < m; i ++) {
ac.insert(br.readLine(), i);
}
br.close();
ac.build();
long ans = ac.dfs(0, 0, 0);
System.out.println(ans);
if(ans <= 42)
ac.print(0, 0, 0);
} catch (IOException e) {
e.printStackTrace();
}
}
}
class AC{
int maxWord = 105; // AC自动机最大记录字母数，最大等于m*字符串最大长度=100，但是还有根节点，不妨设105
int dicLen = 26; // 每一个结点的最大子结点数，即有多少个字母
int maxLen = 30; // 最大密码长度 这里也是个点，不知道为什么写25会出错
int state = (1<<10) + 5; // 1 << M，实际上是一个用二进制表示的状态集，“1”位表示这个词已经出现过了
long[][][] dp = new long[maxWord][maxLen][state]; // 深度搜索、动态规划的矩阵 设d[u][len][state]表示最后一个是字典树的u结点 已选长度为len，状态为state的剩余种数
int[] fail = new int[maxWord]; // 状态转移表
int[] val = new int[maxWord]; // 用一个二进制化的整数表示当前已经出现了多少个词 全0表示当前没有出现词 全1表示字典中的词全出现了
int[] out = new int[dicLen]; // 输出用
public int[][] trie = new int[maxWord][dicLen]; // 字典树
private int nodeIndex = 1; // 当前的下标 0表示根结点
int n;
int m;
public AC(int n, int m) {
this.n = n;
this.m = m;
for(int i = 0; i < maxWord; i ++) { // 对数组全部赋-1，-1表示没有搜索过，0表示搜索过但是没有匹配项
for(int j = 0; j < maxLen; j ++) {
for(int k = 0; k < state; k ++) {
dp[i][j][k] = -1;
}
}
}
}
public void insert(String word, int v) {
int pre = 0;
for (int i = 0; i < word.length(); i ++) {
int code = toInteger(word.charAt(i));
if (trie[pre][code] == 0) { // 如果字典树中不存在这个结点
trie[pre][code] = nodeIndex ++; // 增加一个结点 记录这个值
}
pre = trie[pre][code];
}
val[pre] |= (1 << v); // 标记状态 如果AC自动机经过pre结点，则val[pre]记录的词都出现在了字符串中
}
/**构建next列表*/
public void build() {
ConcurrentLinkedQueue queue = new ConcurrentLinkedQueue();
// java自带的队列就是LinkedList 但LinkedList是线程不安全的 这里必需用线程安全的 因为有递归调用问题
// 把root的所有的子结点加入到队列中，并将这些结点的fail指向根结点
fail[0] = 0;
for (int c = 0; c < dicLen; c ++) {
int u = trie[0][c];
if (u != 0) {
fail[u] = 0;
queue.add(u);
}
}
while (!queue.isEmpty()) {
// 弹出一个结点
int r = queue.poll();
// 遍历下面的每一个结点
for (int c = 0; c < dicLen; c ++) {
int u = trie[r][c];
if (u == 0) { // 如果字典树中不存在当前遍历的结点
trie[r][c] = trie[fail[r]][c]; // 子结点指向父节点失败结点的对应子结点
continue;
}
queue.add(u);
int v = fail[r];
while (v != 0 && trie[v][c] == 0) // 如果还存着fail结点，并且这个结点的对应子结点却不存在，实际就是访问所有fail链
v = fail[v];
fail[u] = trie[v][c]; // 要么存在 要么为0
val[u] |= val[fail[u]]; // 当前遍历这个结点的状态集等于 当前状态集 并上 失败结点指向的状态集
}
}
}
public long dfs(int now, int len, int st) {
if (dp[now][len][st] != -1) // 如果已经搜索过这个结点，直接返回
return dp[now][len][st];
dp[now][len][st] = 0; // 否则，开始搜索这个结点
if (len == n) {
if (st == (1 << m) - 1) // 假设输入的m是3，则1 << m后减1得 000...000111，表示3个词都出现了
return dp[now][len][st] = 1;
return dp[now][len][st] = 0;
}
for (int i = 0; i < dicLen; i++)
dp[now][len][st] += dfs(trie[now][i], len + 1, st|val[trie[now][i]]);
return dp[now][len][st];
}
public void print(int now, int len, int st) {
if (len == n) {
for (int i = 0; i < len ; i ++)
System.out.print((char)(out[i] + 'a'));
System.out.println();
return;
}
for (int i = 0; i < dicLen; i ++) { // 递归查找结果
if (dp[trie[now][i]][len + 1][st|val[trie[now][i]]] > 0) {
out[len] = i;
print(trie[now][i], len + 1, st|val[trie[now][i]]);
}
}
}
public int toInteger(char c) {
return c - 'a';
}
}