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

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

题目来源：599. 两个列表的最小索引总和

2022.03.14 每日一题

LeetCode 题解持续更新中Github仓库地址 CSDN博客地址

今天的题目就比较简单了，题目要求是寻找两个人都喜欢的餐厅，同时还要保证两个人的索引相加起来的和取小值，

这个首先想到的思路就是哈希表，key用于存储String（餐厅名称）value用于存储index（索引）

class Solution {
public:
    vector<string> findRestaurant(vector<string> &list1, vector<string> &list2) {
        // 创建 HashMap 用于存储 list1 中的数据
        unordered_map<string, int> map;
        int len = list1.size();
        // 将 数组 list1 中的数据存入哈希表中
        for (int i = 0; i < len; i++) map[list1[i]] = i;

        // 定义变量统计当前的最小值
        int cnt = 20001;
        len = list2.size();
        // 创建结果数组
        vector<string> list;

        // 遍历数组 list2 ，与 list1 中的元素一一相对比，
        // 如果有相同的就返回 两个索引的加和
        for (int i = 0; i < len; i++) {
            if (map.count(list2[i]) > 0) {
                // 如果找到了更小的，就清空原来的数组，从新找寻，并且更新 cnt 的值
                if (i + map[list2[i]] < cnt) {
                    list.clear();
                    list.push_back(list2[i]);
                    cnt = i + map[list2[i]];
                } else if (i + map[list2[i]] == cnt) {
                    list.push_back(list2[i]);
                }
            }
        }
        // 最后 返回结果
        return list;
    }
};

class Solution {
    public String[] findRestaurant(String[] list1, String[] list2) {
        // 创建 HashMap 用于存储 list1 中的数据
        HashMap<String, Integer> map = new HashMap<>();
        int len = list1.length;
        // 将 数组 list1 中的数据存入哈希表中
        for (int i = 0; i < len; i++) map.put(list1[i], i);
        
        // 定义变量统计当前的最小值
        int cnt = 20001;
        len = list2.length;
        // 创建结果数组
        ArrayList<String> list = new ArrayList<>();
        
        // 遍历数组 list2 ，与 list1 中的元素一一相对比，
        // 如果有相同的就返回 两个索引的加和
        for (int i = 0; i < len; i++) {
            // 如果找到了更小的，就清空原来的数组，从新找寻，并且更新 cnt 的值
            if (i + map.getOrDefault(list2[i], 20001) < cnt) {
                list.clear();
                list.add(list2[i]);
                cnt = i + map.getOrDefault(list2[i], 20001);
            } else if (i + map.getOrDefault(list2[i], 20001) == cnt) {
                list.add(list2[i]);
            }
        }
        // 最后将 ArrayList 转换成为 String[]
        return (String[]) list.toArray();
    }
}