【牛客网】查找兄弟单词

题目链接

---

一个小坑：题目中要求输出第x个兄弟单词。需要判断一下，如果这个数x太大了，就不输出。

package May16;

import java.util.*;

/**
 * 有40%的通过率，我考虑可能是 单词书中不储存 相同的单词？试一下
 * 不是这个原因，错在，题目中要求输出第x个兄弟单词。需要判断一下，如果这个数x太大了，就不输出。
 */
public class Demo11 {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        while (in.hasNext()) {
            int i = in.nextInt();// 保存一共有多少单词
            String[] arr = new String[i];// 保存单词。
//            HashSet<String> set = new HashSet<>();
            for (int j = 0; j < i; j++) {
                String str = in.next();
//                if (!set.contains(str)) {
                arr[j] = str;
//                    set.add(str);
//                }
            }
            Arrays.sort(arr);// 字典排序
            // 成功进行排序
            String toFind = in.next();// 接收字符串
            int toFindInt = in.nextInt();// 接收序号
            int total = 0;
            LinkedList<String> list = new LinkedList<>();
            for (int j = 0; j < i; j++) {
                if (judge(toFind, arr[j])) {
                    total++;
                    list.add(arr[j]);
                }
            }
            System.out.println(total);
            if (toFindInt <= list.size())
            System.out.println(list.get(toFindInt - 1));

        }

    }

    // 判断两个字符串是否是兄弟字符串。
    public static boolean judge(String str1, String str2) {
        if (str1 == null || str2 == null) return false;
        if (str1.length() != str2.length()) {
            return false;
        }
        if (str1.equals(str2)) return false;
        char[] chars1 = str1.toCharArray();
        char[] chars2 = str2.toCharArray();
        Arrays.sort(chars1);
        Arrays.sort(chars2);
        for (int i = 0; i < chars1.length; i++) {
            if (chars1[i] != chars2[i]) {
                return false;
            }
        }
        return true;
    }
}