題目

給定兩個(gè)列表，每個(gè)列表都不含有重復(fù)元素，但兩個(gè)列表之間可能包含有共同元素，請(qǐng)計(jì)算兩個(gè)列表中共同元素的最小索引之和，并以列表的形式，返回對(duì)應(yīng)的元素，如果兩個(gè)列表沒有共同元素，則返回空列表。

例如：

給定兩個(gè)列表：list1=['hello', 'word', '222', 'hi', 'good'], list2=['first', 'hi', 'good', 'WORD', 'hello']
返回結(jié)果：['hi', 'hello']

說明：兩個(gè)列表存在 3 個(gè)共同元素，這3個(gè)元素對(duì)應(yīng)的索引之和分別為 0+4、3+1、4+2，所以最小索引之和為 4 ，對(duì)應(yīng)的重復(fù)元素是 ['hi', 'hello']

給定兩個(gè)列表：list1=['hello', 'word'], list2=['first', 'hi']
返回結(jié)果：[]

說明：兩個(gè)列表不存在共同元素，直接返回空列表。

實(shí)現(xiàn)思路

• 通過set集合的方式求出共同元素，并保存到 tmp_set 中，如果 tmp_set 為空，則直接返回空列表
• 分別用字典 tmp_dict1 和 tmp_dict2 存儲(chǔ)2個(gè)列表中每個(gè)元素及其索引
• 用字典 index_sum_dict 存儲(chǔ)2個(gè)列表中共同元素及對(duì)應(yīng)索引之和
• 求出2個(gè)列表的最小索引之和
• 根據(jù)最小索引之和，求出對(duì)應(yīng)的共同元素

代碼實(shí)現(xiàn)

def findRestaurant(self, list1, list2):
    tmp_set = set(list1) & set(list2)
    if len(tmp_set) == 0:
        return []
    tmp_dict1 = {list1[i]: i for i in range(len(list1))}
    tmp_dict2 = {list2[i]: i for i in range(len(list2))}
    index_sum_dict = {i: (tmp_dict1.get(i) + tmp_dict2.get(i)) for i in tmp_set}
    min_index_sum = min(list(index_sum_dict.values()))
    return [i for i in tmp_set if index_sum_dict.get(i) == min_index_sum]

更多Python編程題，等你來挑戰(zhàn)：Python編程題匯總（持續(xù)更新中……）