599. 两个列表的最小索引总和
假设 Andy 和 Doris 想在晚餐时选择一家餐厅,并且他们都有一个表示最喜爱餐厅的列表,每个餐厅的名字用字符串表示。
你需要帮助他们用最少的索引和找出他们共同喜爱的餐厅。 如果答案不止一个,则输出所有答案并且不考虑顺序。 你可以假设答案总是存在。
题解:
1.先将两个列表的公共元素取出;
2.初始化一个空字典;
3.将公共元素列表中的索引号对应的列表1的索引和列表2中的元素的索引相加放入该字典;
4.求出字典中,键值对中值的最小值,使用filter函数过滤掉不是该最小值的其他值。
class Solution:
def findRestaurant(self, list1: List[str], list2: List[str]) -> List[str]:
aa=set(list1)&set(list2)
res={}
for i in aa:
res[i]=list1.index(i)+list2.index(i)
min_val=min(res.values())
return list(filter(lambda x:res[x]==min_val,res))
备注:
-
filter()函数的简介和语法:
filter()函数用于过滤序列,过滤掉不符合条件的元素,返回符合条件的元素组成新列表。 -
filter()语法如下:
filter(function,iterable) -
序列中的每个元素作为参数传递给函数进行判断,返回True或者False,最后将返回True的元素放到新列表中。
举例:
def is_odd(n):
return n%2 == 1
lst1 = filter(is_odd,[1,2,3,4,5,6,7,8,9,10])
lst1 = [1,3,5,7,9]