Go语言题解LeetCode599两个列表的最小索引总和

时间:2022-12-28 刘09k11 人气:0

题目描述

假设 Andy 和 Doris 想在晚餐时选择一家餐厅，并且他们都有一个表示最喜爱餐厅的列表，每个餐厅的名字用字符串表示。

你需要帮助他们用最少的索引和找出他们共同喜爱的餐厅。如果答案不止一个，则输出所有答案并且不考虑顺序。你可以假设答案总是存在。

示例 1:

输入: list1 = ["Shogun", "Tapioca Express", "Burger King", "KFC"]，list2 = ["Piatti", "The Grill at Torrey Pines", "Hungry Hunter Steakhouse", "Shogun"]
输出: ["Shogun"]
解释: 他们唯一共同喜爱的餐厅是“Shogun”。

示例 2:

输入:list1 = ["Shogun", "Tapioca Express", "Burger King", "KFC"]，list2 = ["KFC", "Shogun", "Burger King"]
输出: ["Shogun"]
解释: 他们共同喜爱且具有最小索引和的餐厅是“Shogun”，它有最小的索引和1(0+1)。

提示:

1 <= list1.length, list2.length <= 1000
1 <= list1[i].length, list2[i].length <= 30
list1[i] 和 list2[i] 由空格 ' ' 和英文字母组成。
list1 的所有字符串都是唯一的。
list2 中的所有字符串都是唯一的。

思路分析

其实题目意思不是找有不有共同的，而是找共同喜欢且索引和最小的，换种理解方式，2个朋友想找个都喜欢的餐馆聚餐，2个人喜欢的餐馆按照距离2个人距离排序，想找一个离2个人都近的餐馆，这样理解就好理解了

所以思路是：

现存一个人的餐馆表，key是餐馆，value是距离

去匹配第二个人的餐馆表，找到了看是不是最小距离，是的话就清空之前的添加进去，相等则添加到列表

AC 代码

class Solution {
    public String[] findRestaurant(String[] list1, String[] list2) {
            if(list1 == null||list2 == null){
                return null;
            }
            HashMap<String,Integer> bucket = new HashMap<>();
            for(int i = 0;i < list1.length;i++){
                bucket.put(list1[i],i);
            }
            List<String> res = new ArrayList<>();
            int minIndexSum = Integer.MAX_VALUE;
            for(int i = 0;i< list2.length;i++){
                if(bucket.containsKey(list2[i])){
                    if(bucket.get(list2[i])+i<minIndexSum){
                        minIndexSum = bucket.get(list2[i])+i;
                        res.clear();
                        res.add(list2[i]);
                    }else if(bucket.get(list2[i])+i == minIndexSum){
                        res.add(list2[i]);
                    }
                }
            }
            return res.toArray(new String[res.size()]);
    }
}

参考

4行代码，思路简单，性能高效 - 两个列表的最小索引总和 - 力扣（LeetCode） (leetcode-cn.com)

加载全部内容