599. 两个列表的最小索引总和（Python）

题目

难度：★★☆☆☆
类型：数组，益智

假设Andy和Doris想在晚餐时选择一家餐厅，并且他们都有一个表示最喜爱餐厅的列表，每个餐厅的名字用字符串表示。

你需要帮助他们用最少的索引和找出他们共同喜爱的餐厅。 如果答案不止一个，则输出所有答案并且不考虑顺序。 你可以假设总是存在一个答案。

提示
两个列表的长度范围都在 [1, 1000]内。
两个列表中的字符串的长度将在[1，30]的范围内。
下标从0开始，到列表的长度减1。
两个列表都没有重复的元素。

示例

示例 1:
输入:
["Shogun", "Tapioca Express", "Burger King", "KFC"]
["Piatti", "The Grill at Torrey Pines", "Hungry Hunter Steakhouse", "Shogun"]
输出: ["Shogun"]
解释: 他们唯一共同喜爱的餐厅是“Shogun”。

示例 2:
输入:
["Shogun", "Tapioca Express", "Burger King", "KFC"]
["KFC", "Shogun", "Burger King"]
输出: ["Shogun"]
解释: 他们共同喜爱且具有最小索引和的餐厅是“Shogun”，它有最小的索引和1(0+1)。

解答

题意不仅要求找到Andy和Doris两个人都提议的饭店，而且还要这个饭店在两个人的列表中出现的位置和最小。我们可以这样考虑：

1. 建立两个字典，分别代表Andy和Doris的列表中的元素及其所在的位置，字典的键是饭店名，字典的值是该饭店名出现在Andy（或Doris）的列表中的位置；

2. 求取两人列表的交集，包含了两个人均提议的饭店，我们最终所求其实是这个交集的一个子集；

3. 再次建立一个字典，用来表示饭店与下标和之间的关系，字典的键是出现在上述交集中的饭店，字典的值是该饭店在两个人列表中的位置的和，并求出该字典中的最小位置和（该字典值的最小值）；

4. 对于交集中的每一个饭店，在字典中筛选出值为最小位置和的饭店，将筛选后的结果存放在新的列表中返回。

class Solution:
    def findRestaurant(self, list1, list2):
        
        # 获得字典index1：表示list1中各个元素与其下标的对应关系
        index1 = {list1[i]: i for i in range(len(list1))}
        
        # 获得字典index2：表示list2中各个元素与其下标的对应关系
        index2 = {list2[i]: i for i in range(len(list2))}
        
        # 获取集合list1和list2的交集
        agreements = set(list1) & set(list2)
        
        # 获得字典sum_index，表示交集中每个元素与其在两个list中下标和的对应关系
        sum_index = {r: index1[r]+index2[r] for r in agreements}
        
        # 选出交集中下标和最小的元素
        return [r for r in agreements if sum_index[r] == min(sum_index.values())]

只需要五行代码实现复杂功能，是python的一大魅力~

如有疑问或建议，欢迎评论区留言~