怎么用python实现:题目描述 一个单词a如果通过交换单词中字母的顺序可以得到另外的单词b,那么定义b是a的Anigram,例如单词army和mary互为Anigram。现在给定一个字典,输入Q个单词,从给出的字典中找出这些单词的Anigram。 要求该功能用函数实现,传含Q个单词的字典,传单词,在字典中找出单词互为Anigram的单词个数,并返回结果(单词不与自身互为Anigram)。 输入 第1行:1个数N,表示字典中单词的数量。(1 <= N <= 10000) 第2 - N + 1行,字典中的单词,单词长度 <= 10。 第N + 2行:查询的数量Q。(1 <= Q <= 10000) 第N + 3 - N + Q - 2行:用作查询的单词,单词长度 <= 10。 输出 共Q行,输出Anigram的数量,相同的2个单词不算Anigram,如果没有输出0。
时间: 2024-03-07 15:48:19 浏览: 18
可以使用字典来实现,将字典中的每个单词按照字母顺序排序后,作为key存储在字典中,对于每个查询单词,也按照字母顺序排序后进行查询,找到相同的即为Anigram。
具体实现代码如下:
```python
def find_anigrams(word_dict, query_word):
sorted_word = ''.join(sorted(query_word))
if sorted_word in word_dict:
return word_dict[sorted_word]
else:
return 0
def count_anigrams(word_list, query_list):
word_dict = {}
count = 0
for word in word_list:
sorted_word = ''.join(sorted(word))
if sorted_word not in word_dict:
word_dict[sorted_word] = 1
else:
word_dict[sorted_word] += 1
for query in query_list:
if query in word_dict:
count += word_dict[query] - 1
else:
count += find_anigrams(word_dict, query) - 1
return count
if __name__ == '__main__':
n = int(input())
word_list = []
for i in range(n):
word_list.append(input().strip())
q = int(input())
query_list = []
for i in range(q):
query_list.append(input().strip())
ans = count_anigrams(word_list, query_list)
print(ans)
```
其中 `word_dict` 为存储字典中单词的字典,键为单词按照字母顺序排序后的字符串,值为该单词在字典中出现的次数。在 `count_anigrams` 函数中,首先遍历字典中的单词,将它们按照字母顺序排序后存储在 `word_dict` 中。然后遍历查询单词,对于每个查询单词,如果它在 `word_dict` 中出现过,那么将 `word_dict` 中该单词出现的次数减去1加到 `count` 中,因为相同的2个单词不算Anigram;否则,就调用 `find_anigrams` 函数在 `word_dict` 中找到与查询单词为Anigram的单词个数,将结果减去1加到 `count` 中。最后输出 `count` 即为结果。
阅读全文