题目大意:就是给一串电话号码,问是否存在这种情况,一个号码是另一个号码的前缀!!!
解题思路:字典树(tire树),是一种极大节省空间的树形结构,因为所有具有相同前缀的会共享前缀,我对tire数的理解是,结点本身是不存信息,存信息是当前结点的某个子节点是否为空!!!
tire树的结构如下:
而tire树的核心代码归为两个函数,createTire()和findTire(),就像并查集的核心就是find()、inti()和union():
那我们想想tire的结构体该怎么写呢?
由于我们要判断一个集合里是否存在谁是谁的前缀,而我们是用树形的结构来存储的,那相应的,我们应该在tire的结构体中设置一个标志位,来记录当前结点是不是一个字符串的最后一个。
这一题我们可以把create()和find()一起进行!!!
上码:
import java.io.BufferedInputStream;
import java.util.Scanner;
public class Main {
static Tire root;
public static void main(String[] args) {
Scanner sc = new Scanner(new BufferedInputStream(System.in));
int cases, n;
String input;
boolean flag;// 标志是否有前缀
cases = sc.nextInt();
while (cases-- > 0) {
n = sc.nextInt();
flag = false;// 默认无前缀
root = new Tire();
while (n-- > 0) {
input = sc.next();
if (!flag) {<span style="font-family: Arial, Helvetica, sans-serif;">//</span><span style="font-family: 宋体;">只要发现已经有前缀了,接下来的就不需要再判断了</span>
if (findAndCreate(input)) {
flag = true;
}
}
}
if (flag) {
System.out.println("NO");
} else {
System.out.println("YES");
}
}
sc.close();
}
// true:找到了,false:没有找到
static boolean findAndCreate(String str) {
Tire p = root;
for (int i = 0; i < str.length(); i++) {
int id = str.charAt(i) - '0';
if (p.next[id] != null && p.next[id].isEnd) {// 树中已经有当前串的前缀了
return true;
} else if (p.next[id] == null) {// 树中没有当前串,接着新建
p.next[id] = new Tire();
}
p = p.next[id];
}
p.isEnd=true;
for (Tire t : p.next) {<span style="color: rgb(63, 127, 95); font-family: Arial, Helvetica, sans-serif;">// 当前串是树的前缀</span>
if (t != null) {
return true;
}
}
return false;
}
}
class Tire {
Tire[] next;
boolean isEnd;<span style="font-family: 宋体; font-size: 12pt;">//</span><span style="font-family: 宋体; font-size: 12pt;">是否是一个串的结束的最后一个字符</span>
Tire() {
next = new Tire[10];
}
}