599. 两个列表的最小索引总和『简单』

题目来源于力扣（LeetCode）

目录

• 一、题目
• 二、解题思路
• 三、代码实现
• 四、执行用时
• 五、部分测试用例

一、题目

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

题目相关标签：哈希表

提示：

1. 两个列表的长度范围都在 [1, 1000]内。
2. 两个列表中的字符串的长度将在[1，30]的范围内。
3. 下标从0开始，到列表的长度减1。
4. 两个列表都没有重复的元素。

二、解题思路

1. 创建 map 集合记录元素值及其索引，创建 list 集合用于存储结果集，声明变量用于记录最小的索引和

2. 遍历 list1 数组，将 list1 数组中的元素作为键，元素所在的索引作为值存储到 map 集合中

3. 遍历 list2 数组，判断当前遍历的元素是否存在于 list1 数组中，且索引位与该元素在 list1 数组中的索引位相加小于等于最小的索引和

4. 大于时，不做操作；等于时，说明存在多个结果，存储到 list 集合中

5. 小于时，即出现更小的索引和，需要清空 list 集合中的元素并记录下该索引和，当前遍历的元素添加到 list 中

三、代码实现

public static String[] findRestaurant(String[] list1, String[] list2) {
    List<String> list = new ArrayList<>();
    // map 记录 list1 数组中元素出现的索引位
    Map<String, Integer> map = new HashMap<>();
    // 记录索引和，初始为能达到的最大索引和
    int minIndex = list1.length + list2.length;

    for (int i = 0; i < list1.length; i++) {
        map.put(list1[i], i);
    }
    for (int i = 0; i < list2.length; i++) {
        if (map.containsKey(list2[i])) {
            // 计算出相同元素的索引和
            int indexSum = map.get(list2[i]) + i;
            if (indexSum > minIndex) {
                continue;
            }
            // 最小和出现时
            else if (indexSum < minIndex) {
                if (list.size() > 0) {
                    // 清空 list 数组
                    list.clear();
                }
                // 重置最小和
                minIndex = indexSum;
            }
            list.add(list2[i]);
        }
    }
    return list.toArray(new String[list.size()]);
}

四、执行用时

五、部分测试用例

public static void main(String[] args) {
    String[] list1 = {"Shogun", "Tapioca Express", "Burger King", "KFC"};
    String[] list2 = {"Piatti", "The Grill at Torrey Pines", "Hungry Hunter Steakhouse"};
    // output:{"Shogun"}

//    String[] list1 = {"Shogun", "Tapioca Express", "Burger King", "KFC"};
//    String[] list2 = {"KFC", "Shogun", "Burger King"};
    // output:{"Shogun"}
    
//    String[] list1 = {"Shogun", "KFC", "Burger King"};
//    String[] list2 = {"KFC", "Shogun", "Burger King"};
    // output:{"Shogun", "KFC"}  索引和都是 1
    String[] result = findRestaurant(list1, list2);
    System.out.println(Arrays.toString(result));
}