Python算法--查找兄弟单词

题目

输入描述：

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

输出描述：

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

答案与解析

while True:
    try:
        s = input().split()
        dic = s[1:-2]
        x = s[-2]
        k = int(s[-1])
        brother = []

        for i in dic:
            if sorted(i)==sorted(x) and i!=x:
                brother.append(i)     
        brother =  sorted(brother)
        print(len(brother))
        print(brother[k-1])
    except:
        break