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),节约空间