JAVA 优先队列 兄弟单词快速判断| #查找兄弟单词#
查找兄弟单词
https://www.nowcoder.com/practice/03ba8aeeef73400ca7a37a5f3370fe68
import java.util.*;
// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
public static boolean isBro(String X, String n){
if(n.length()!=X.length())return false;
if(n.equals(X))return false;
for(char c : X.toCharArray()){
if(!(n.replace(c+"","").length()==X.replace(c+"","").length())){
return false;
}
}
return true;
}
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
String line = in.nextLine();
String[] line_split = line.split(" ");
int k = Integer.valueOf(line_split[line_split.length-1]);
PriorityQueue<String> pq = new PriorityQueue<>((s, t1) -> -s.compareTo(t1));
String X = line_split[line_split.length-2];
int bro_num = 0;
for(int i=1;i<line_split.length-2;i++){
String n = line_split[i];
if(isBro(X,n)){
bro_num++;
pq.add(n);
}
if(pq.size()>k) pq.poll();
}
System.out.println(bro_num);
if(pq.size()==k) System.out.println(pq.peek());
}
}
第k个:优先队列算法节约空间
兄弟单词:
可以使用int[26] 存储字符个数判断
这里使用替换后判断大小O(n),节约空间
