LeetCode 599 两个列表的最小索引总和[map] HERODING的LeetCode之路

Posted 2022-03-16 HERODING23

解题思路：
一道非常简单的map应用题目，用一个map存储其中一个list的所有餐厅名称和其序号，因为最后反悔哦结果中的餐厅肯定在该map中，所以只用一个map，接着遍历另一个list，一旦找到餐厅在map中，就计算其索引和，如果最小，更新最小值并将数组重置，如果和最小值相同，放入数组后面即可，代码如下：

class Solution 
public:
    vector<string> findRestaurant(vector<string>& list1, vector<string>& list2) 
        vector<string> ans;
        unordered_map<string, int> mp;
        for(int i = 0; i < list1.size(); i ++) 
            mp[list1[i]] = i; 
        
        int love = INT_MAX;
        for(int i = 0; i < list2.size(); i ++) 
            if(mp.count(list2[i]) > 0) 
                int cur = i + mp[list2[i]];
                if(cur < love) 
                    ans.clear();
                    love = cur;
                    ans.push_back(list2[i]);
                 else if(cur == love) 
                    ans.push_back(list2[i]);
                
            
        
        return ans;
    
;