HJ27 查找兄弟单词

https://www.nowcoder.com/practice/03ba8aeeef73400ca7a37a5f3370fe68 (opens new window)

描述

定义一个单词的“兄弟单词”为：交换该单词字母顺序（注：可以交换任意次），而不添加、删除、修改原有的字母就能生成的单词。 兄弟单词要求和原来的单词不同。例如： ab 和 ba 是兄弟单词。 ab 和 ab 则不是兄弟单词。 现在给定你 n 个单词，另外再给你一个单词 x ，让你寻找 x 的兄弟单词里，按字典序排列后的第 k 个单词是什么？ 注意：字典中可能有重复单词。

输入描述：

输入只有一行。 先输入字典中单词的个数n，再输入n个单词作为字典单词。 然后输入一个单词x 最后后输入一个整数k

输出描述：

第一行输出查找到x的兄弟单词的个数m 第二行输出查找到的按照字典顺序排序后的第k个兄弟单词，没有符合第k个的话则不用输出。

示例1：

输入：
3 abc bca cab abc 1

输出：
2
bca

示例2：

输入：
6 cab ad abcd cba abc bca abc 1

输出：
3
bca

说明：
abc的兄弟单词有cab cba bca，所以输出3
经字典序排列后，变为bca cab cba，所以第1个字典序兄弟单词为bca      

解答

const rl = require("readline").createInterface({ input: process.stdin });
var iter = rl[Symbol.asyncIterator]();
const readline = async () => (await iter.next()).value;

void async function () {
    let line;
    let words;
    let targetWord;
    let index;

    while(line = await readline()){
        let tokens = line.split(' ');
        index = parseInt(tokens.pop());
        targetWord = tokens.pop();
        
        words = tokens.slice(1);
    }

    // console.log(words, targetWord, index)

    const brotherWords = [];

    const targetWordLen = targetWord.length;

    const sortedTargetWord = sort(targetWord);

    words.forEach((word) => {
        if (word.length !== targetWordLen) {
            return;
        }

        if (word === targetWord) {
            return;
        }

        const sortedWord = sort(word);

        // 字母排序后，如果两单词一样，则是兄弟单词
        if (sortedTargetWord === sortedWord) {
            brotherWords.push(word);
        }
    });

    const sortedBrotherWords = brotherWords.sort();

    const brotherWordsLen = brotherWords.length;
    const brotherWord = sortedBrotherWords[index - 1];

    console.log(brotherWordsLen);

    if (brotherWord) {
        console.log(brotherWord);
    }
}()

function sort(word) {
    return word
        .split('')
        .sort((char1, char2) => {
            if (char1 > char2) {
                return 1;
            } else if (char1 < char2) {
                return -1;
            } else {
                return 0;
            }
        })
        .join('')
}