LeetCode算法题-Minimum Index Sum of Two Lists（Java实现）

这是悦乐书的第272次更新，第286篇原创

01 看题和准备

今天介绍的是LeetCode算法题中Easy级别的第139题（顺位题号是599）。假设Andy和Doris想要选择一家餐馆吃晚餐，他们都有一个最受欢迎的餐馆列表。你需要用最少的列表索引总和帮助他们找出他们的共同兴趣。如果答案之间存在选择关系，则输出所有答案并且没有顺序要求。你可以假设总有一个答案。例如：

输入:

["Shogun", "Tapioca Express", "Burger King", "KFC"]

["Piatti", "The Grill at Torrey Pines", "Hungry Hunter Steakhouse", "Shogun"]

输出: ["Shogun"]

说明: 他们都喜欢的唯一一家餐厅是“Shogun”。


输入:

["Shogun", "Tapioca Express", "Burger King", "KFC"]

["KFC", "Shogun", "Burger King"]

输出: ["Shogun"]

说明: 他们喜欢并且索引总和最少的餐馆是“Shogun”，索引和1（0 + 1）。

注意：

• 两个列表的长度将在[1,1000]的范围内。

• 两个列表中的字符串长度将在[1,30]的范围内。

• 索引从0开始到列表长度减去1。

• 两个列表中都没有重复项。

本次解题使用的开发工具是eclipse，jdk使用的版本是1.8，环境是win7 64位系统，使用Java语言编写和测试。

02 第一种解法

使用两个HashMap，分别将list1和list2的元素存入其中，以元素值为key，索引为value。定义一个最小值min，初始值为int的最大值，然后遍历其中一个map的key值，如果第二个map中也存在该key，此时分为两种情况处理。

第一种情况，如果该key值对应两索引之和小于min，那么将min更新，再将结果数组清空，再将该key添加进结果数组。

第二种情况，如果该key值对应两索引之和等于min，表明存在另外一对索引之和，直接将key添加进结果数组即可。

最后返回结果数组。

public String[] findRestaurant(String[] list1, String[] list2) {
    Map<String, Integer> map = new HashMap<>();
    Map<String, Integer> map2 = new HashMap<>();
    for(int i=0; i<list1.length; i++){
        map.put(list1[i], i);
    }
    for(int i=0; i<list2.length; i++){
        map2.put(list2[i], i);
    }
    int min = Integer.MAX_VALUE;
    List<String> result = new ArrayList<>();
    for (String key : map.keySet()) {
        if (map2.containsKey(key)) {
            if (map.get(key)+map2.get(key) < min) {
                min = map.get(key)+map2.get(key);
                result = new ArrayList<>();
                result.add(key);
            } else if (map.get(key)+map2.get(key) == min) {
                result.add(key);
            }
        }
    }
    return result.toArray(new String[result.size()]);
}

03 第二种解法

我们可以将第一种解法再简化下，只使用一个HashMap，然后遍历剩下的那个数组，判断思路与第一种解法一致。同时，我们也可以将循环内部使用创建新对象来清空数组的方法，换成其自身的clear()方法。

public String[] findRestaurant2(String[] list1, String[] list2) {
    Map<String, Integer> map = new HashMap<>();
    for(int i=0; i<list1.length; i++){
        map.put(list1[i], i);
    }
    int min = Integer.MAX_VALUE;
    List<String> result = new ArrayList<>();
    for (int i=0; i<list2.length; i++) {
        if (map.containsKey(list2[i])) {
            if (map.get(list2[i])+i < min) {
                min = map.get(list2[i])+i;
                result.clear();
                result.add(list2[i]);
            } else if (map.get(list2[i])+i == min) {
                result.add(list2[i]);
            }
        }
    }
    return result.toArray(new String[result.size()]);
}

04 第三种解法

我们也可以不使用HashMap，直接使用for循环来解决。

public String[] findRestaurant3(String[] list1, String[] list2) {
    int min = Integer.MAX_VALUE;
    List<String> result = new ArrayList<>();
    for (int i=0; i<list1.length; i++) {
        for (int j=0; j<list2.length; j++) {
            if (list1[i].equals(list2[j])) {
                if (j+i < min) {
                    min = j+i;
                    result.clear();
                    result.add(list1[i]);
                } else if (j+i == min) {
                    result.add(list1[i]);
                }
            }
        }
    }
    return result.toArray(new String[result.size()]);
}

05 小结

算法专题目前已日更超过四个月，算法题文章139+篇，公众号对话框回复【数据结构与算法】、【算法】、【数据结构】中的任一关键词，获取系列文章合集。

以上就是全部内容，如果大家有什么好的解法思路、建议或者其他问题，可以下方留言交流，点赞、留言、转发就是对我最大的回报和支持！