LeetCode 每日一题：599. 两个列表的最小索引总和

最新推荐文章于 2022-03-14 20:37:05 发布

原创最新推荐文章于 2022-03-14 20:37:05 发布 · 554 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #散列表 #哈希算法 #c++ #算法

每日一题专栏收录该内容

144 篇文章

订阅专栏

该代码实现了一个寻找两个列表中元素匹配时，其下标和最小的餐厅名称的算法。通过哈希映射存储一个列表的元素及其下标，然后遍历另一个列表，计算下标和，找到和最小的元素并更新答案。

599. 两个列表的最小索引总和

首先通过哈希存储其中一个列表下标，遍历另一个列表每次比对答案下标之和，如果相等直接加到答案里面，如果更小那么清空之前的答案，同时更新最小下标和

class Solution {
public:
    vector<string> findRestaurant(vector<string>& list1, vector<string>& list2) {
        map<string, int> mp;
        for (int i = 0; i < list1.size(); i ++ )
            mp[list1[i]] = i;
        
        vector<string> ans;
        int minn = INT_MAX;
        for (int j = 0; j < list2.size(); j ++ ) {
            string s = list2[j];
            if (mp.count(s)) {
                if (mp[s] + j < minn) {
                    ans.clear();
                    ans.push_back(s);
                    minn = mp[s] + j;
                }
                else if (mp[s] + j == minn) {
                    ans.push_back(s);
                }
            }
        }
        return ans;
    }
};