Leetcode 599.兩個列表的最小索引總和
1 題目描述(Leetcode題目鏈接)
假設Andy和Doris想在晚餐時選擇一家餐廳,並且他們都有一個表示最喜愛餐廳的列表,每個餐廳的名字用字符串表示。
你需要幫助他們用最少的索引和找出他們共同喜愛的餐廳。 如果答案不止一個,則輸出所有答案並且不考慮順序。 你可以假設總是存在一個答案。
輸入:
["Shogun", "Tapioca Express", "Burger King", "KFC"]
["Piatti", "The Grill at Torrey Pines", "Hungry Hunter Steakhouse", "Shogun"]
輸出: ["Shogun"]
解釋: 他們唯一共同喜愛的餐廳是“Shogun”。
輸入:
["Shogun", "Tapioca Express", "Burger King", "KFC"]
["KFC", "Shogun", "Burger King"]
輸出: ["Shogun"]
解釋: 他們共同喜愛且具有最小索引和的餐廳是“Shogun”,它有最小的索引和1(0+1)。
提示:
- 兩個列表的長度範圍都在 [1, 1000]內。
- 兩個列表中的字符串的長度將在[1,30]的範圍內。
- 下標從0開始,到列表的長度減1。
- 兩個列表都沒有重複的元素。
2 題解
就是求交集中下標和最小的。
class Solution:
def findRestaurant(self, list1: List[str], list2: List[str]) -> List[str]:
res = []
min_val = float("inf")
dic = {}
for i in range(len(list2)): dic[list2[i]] = i
for i in range(len(list1)):
if list1[i] in dic:
if i + dic[list1[i]] < min_val:
res = [list1[i]]
min_val = i + dic[list1[i]]
elif i + dic[list1[i]] == min_val:
res.append(list1[i])
return res