-(题目搜索牛客网)
编程中的问题
超时
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
import org.junit.Test;
public class TheShortestPrefix {
@SuppressWarnings("resource")
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
while(scanner.hasNext()) {
int num = scanner.nextInt();
String[] strs = new String[num];
for (int i = 0; i < num; i++) {
strs[i] = scanner.next();
}
//获取有效的前缀
String[] prefixs = findshortestprefix(strs);
for (String string : prefixs) {
System.out.println(string);
}
//输入空格
System.out.println();
}
}
public static String[] findshortestprefix(String[] strs) {
String[] vaildPrefix = new String[strs.length];
int i = 0;
for (i =0;i<strs.length;i++) {
// System.out.println("str:"+strs[i]);
List<String> prefixs = createprefix(strs[i]);
for (String prefix : prefixs) {
// System.out.println("str :"+strs[i]+" prefix :"+prefix+":"+vaild(strs, prefix, strs[i]));
//如果前缀合法,而且排列是从小到大,直接取该合法前缀
if(vaild(strs, prefix, strs[i])) {
vaildPrefix[i] = prefix;
break;
}
vaildPrefix[i] = strs[i];
}
System.out.println();
}
return vaildPrefix;
}
//所有的前缀
private static List<String> createprefix(String str) {
ArrayList<String> list = new ArrayList<String>();
for(int i=0;i<str.length();i++) {
list.add(str.substring(0, i+1));
}
return list;
}
//判断前缀合法
public static boolean vaild(String[] strs ,String prefix,String parent) {
//如果str的前缀和其他字符串的前缀重合不合法
for (String string : strs) {
if(!string.equals(parent)) {
//获取其他字符串的前缀数组做比较
List<String> list = createprefix(string);
for (String p : list) {
if(p.equals(prefix))return false;
}
}
}
return true;
}
}
网上查询到的解决途径
使用了前缀树来解决
这是我后来写的代码,运行时间果然变短了,运行通过
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
import org.junit.Test;
public class TheShortestPrefix {
@SuppressWarnings("resource")
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
while(scanner.hasNext()) {
int num = scanner.nextInt();
String[] strs = new String[num];
for (int i = 0; i < num; i++) {
strs[i] = scanner.next();
}
Trie root = createTrie(strs);
for (String s : strs) {
String prefix = prefix(root, s);
System.out.println(prefix);
}
//空格
System.out.println();
}
}
//1. 创建一个前缀树,默认是每个节点有26子节点
public static class Trie {
private char name;
private int num = 1;
private Trie[] next = new Trie[26];//26字母
//构造函数
public Trie(char name) {
this.name = name;
}
}
//2. 创建一个将String数组结构转变成一个前缀树并返回生成的前缀树的方法
public static Trie createTrie(String[] str) {
//2.1 首先生成一个root,内容无所谓
Trie root = new Trie('-');
Trie cur;//中间值
//2.2 遍历数组将字符插入这个前缀树中
for (String s : str) {
cur = root;
for (int i = 0; i < s.length(); i++) {
//2.3 插入字符
int position = s.charAt(i)-'a';//从0开始到25
if(cur.next[position]==null) {
cur.next[position] = new Trie(s.charAt(i));
}else {
cur.next[position].num++;
}
//2.4 进入下一个循环
cur = cur.next[position];/*这里自己少写了*/
}
}
//2.3 返回前缀树
return root;
}
//3. 查询一个包含在一个数组的字符串在该数组的前缀树中前缀并返回该前缀字符串
public static String prefix(Trie root,String str) {
int len = 0;//前缀的长度
Trie cur;
/*
* 注意 三个if else 条件 第一个是返回空字符串,第三个是不符合进行下一循环,第二步才是找到最短前缀结束
*/
for (int i = 0; i < str.length(); i++) {
//3.1 获取当前节点下一个位置的节点
cur = root.next[str.charAt(i)-'a'];/*这里自己循环修改了,原来写错了*/
if(cur==null) {
break;
}else if(cur.num==1) {
//3.2 当前符合
len++;/*这里自己少写了*/
break;
}else {
len++;//和别的字符串重复不符合,查下一个
//3.2 下一步将当前节点换成如下
root = cur;
}
}
return str.substring(0, len);
}
/*//测试用例
@Test
public void test() {
String[] str= {"ab","a","acb"};
Trie t = createTrie(str);
String prefix = prefix(t, "acb");
System.out.println(prefix);
}*/
}