查找兄弟单词

题目描述

输入描述:

先输入字典中单词的个数，再输入n个单词作为字典单词。
输入一个单词，查找其在字典中兄弟单词的个数
再输入数字n

输出描述:

根据输入，输出查找到的兄弟单词的个数

输入：

3    abc    bca    cab    abc    1

输出：

2    bca

代码如下：

def word_bro():
    s = input().split()
    word_count = int(s[0])
    word_list = s[1:word_count+1]
    word_list.sort()
    word_need = s[-2]
    word_index = int(s[-1])
    word_bro_list = []
 
    word_need_list = [b for b in word_need]
    word_need_list.sort()
 
    for word in word_list:
        word_cmp_list = [a for a in word]
        word_cmp_list.sort()
        if word != word_need and word_cmp_list == word_need_list:
                word_bro_list.append(word)
                 
 
    print(len(word_bro_list))
    if len(word_bro_list) > 0 and word_index-1 <= len(word_bro_list):
        print(word_bro_list[word_index-1])
 
while True:
    try:
        word_bro()
    except:
        break