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

全部评论

相关推荐

1 收藏 评论
分享
牛客网
牛客企业服务