Leetcode 599.兩個列表的最小索引總和(Minimum Index Sum of Two Lists)

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
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章