1. 程式人生 > >Leetcode599.Minimum Index Sum of Two Lists

Leetcode599.Minimum Index Sum of Two Lists

假設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 {
public:
    vector<string> findRestaurant(vector<string>& list1, vector<string>& list2) {
        map<string, int> hax;
        int len1 = list1.size();
        int len2 = list2.size();
        for(int i = 0; i < len1; i++)
        {
            hax[list1[i]] = i + 1;
        }
        vector<string> res;
        int MIN = INT_MAX;
        for(int i = 0; i < len2; i++)
        {
            if(hax[list2[i]] > 0 && i + hax[list2[i]] - 1 < MIN)
            {
                res.clear();
                MIN = i + hax[list2[i]] - 1;
            }
            if(hax[list2[i]] > 0 && i + hax[list2[i]] - 1 == MIN)
            {
                res.push_back(list2[i]);
            }
        }
        return res;
    }
};