LeetCode-599

两个列表的最小索引总和

题目
假设Andy和Doris想在晚餐时选择一家餐厅，并且他们都有一个表示最喜爱餐厅的列表，每个餐厅的名字用字符串表示。
帮助他们用最少的索引和找出他们共同喜爱的餐厅。 如果答案不止一个，则输出所有答案并且不考虑顺序。

示例
输入:
[“Shogun”, “Tapioca Express”, “Burger King”, “KFC”]
[“Piatti”, “The Grill at Torrey Pines”, “Hungry Hunter Steakhouse”, “Shogun”]
输出: [“Shogun”]
解释: 他们唯一共同喜爱的餐厅是“Shogun”。

题解

方法 1 哈希表（线性）

本题需要找出最小索引和的数组元素交集，可以遍历其中一个数组，对每一个元素，在另一个数组中进行查找，若存在则计算索引和，如果索引和更小则更新结果为当前元素，如果索引和不变则向结果中加入当前元素，如果索引和更大则不变。由于数组的查找操作耗时较多，故可以利用哈希表存储数组元素及其对应下标，便于查找。

更多题解

代码

class Solution {
public:
    vector<string> findRestaurant(vector<string>& list1, vector<string>& list2) {
        unordered_map<string,int> map;
        int m=list1.size(),n=list2.size();
        vector<string> res;
        //索引和范围[0,m+n-2]
        int min_sum=m+n-2;
        //哈希表存储数组list1
        for(int i=0;i<m;++i){
            map[list1[i]]=i;
        }
        for(int j=0;j<n;++j){
        	//判断map中是否存在当前元素
        	//若不存在则跳过本次循环，进行下一次循环
        	//若存在则继续执行后续语句
            if(!map.count(list2[j])){
                continue;
            }
            //若索引和不变则将当前元素加入结果
            //若索引和更小则清空，将当前元素加入结果
            if(map[list2[j]]+j==min_sum){
                res.emplace_back(list2[j]);
            }else if(map[list2[j]]+j<min_sum){
                min_sum=map[list2[j]]+j;
                res.clear();
                res.emplace_back(list2[j]);
            }
        }
        return res;
    }
};