1143. 两个列表的最小索引和

1143. 两个列表的最小索引和

中文English

假设Andy和Doris想要选择一家餐馆吃晚餐，他们都有一个各自最喜爱的餐馆列表。

你需要用最少的列表索引总和来帮助他们找出他们的共同兴趣。 如果最少列表索引总和的答案不唯一，则输出所有答案并且没有顺序要求。 你可以假设总有一个答案。

样例

案例 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).

注意事项

1.两个列表的长度都将在[1,1000]的范围内。
2.两个列表中的字符串长度将在[1,30]范围内。
3.索引从0开始直到到列表长度减1。
4.两个列表中都没有重复。

输入测试数据 (每行一个参数)如何理解测试数据？

##最小索引和
class Solution:
    ''' 
    大致思路:
    1.初始化dic字典，{共同餐馆:索引和...}，给出初始最小索引和min_index_sum = len(list1) + len(list2)，给定min_index_sum,current_value初始化.
    2.循环list1，判断元素是否存在list2里面，如果存在，则给出列表索引和,同时要判断索引和是否小于min_index_sum
    如果符合的话，给出一个暂存空间，先不存储进dic里面
    此时分两种情况:
    (1).如果出现相同的列表索引和，则存储上一次的元素和列表索引和，此时current_value，和min_index_sum重新赋值，但不存储到dic里面
    (2).如果出现更小的列表索引和，此时current_value，和min_index_sum重新赋值，但不存储到dic里面
    最后把current_value，和min_index_sum赋值给dic，最后循环dic的index，append到res里面，返回
    '''
    def findRestaurant(self,list1,list2):
        dic,res = {},[]
        min_index_sum,current_value = len(list1) + len(list2),''
        for i in range(len(list1)):
            if list1[i]  in list2:
                if  i + list2.index(list1[i]) <= min_index_sum:
                    #如果是等于的话，就把上一个加进来
                    if i + list2.index(list1[i]) == min_index_sum:
                        dic[current_value] = min_index_sum

                    #给一个暂存的变量，存储的是上一个共同的元素
                    current_value = list1[i]
                    min_index_sum = i + list2.index(list1[i])
       
        #最后在把这个值加进来（因为每次存储的都是上一个值，所以如果只有一个的话，那么上面并没有存储进来。如果存在两个相同的话，那么上面只存储了一次，还需要加进来)
        dic[current_value] = min_index_sum
        return [index for index in dic.keys()]

不用dic字典写法:

class Solution:
    def findRestaurant(self,list1,list2):
        res = []
        s,current_value = len(list1) + len(list2),''
        for i in range(len(list1)):
            if list1[i]  in list2:
                if  i + list2.index(list1[i]) <= s:
                    #如果是等于的话，就把上一个加进来
                    if i + list2.index(list1[i]) == s:
                        res.append(current_value)

                    #给一个暂存的变量，存储的是上一个共同的元素
                    current_value = list1[i]
                    
                    s = i + list2.index(list1[i])
       
        #最后在把这个值加进来（因为每次存储的都是上一个值，所以如果只有一个的话，那么上面并没有存储进来。如果存在两个相同的话，那么上面只存储了一次，还需要加进来)
        res.append(current_value)
        return res

result = Solution().findRestaurant(["Shogun", "Tapioca Express", "Burger King", "KFC"],["KFC", "Shogun", "Burger King"])
print(result)