【leetcode15-----两个列表的最小索引总和】

两个列表的最小索引总和

假设 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)。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/minimum-index-sum-of-two-lists
 

题解：

题目讲的有些不太明白，其实题目要求的是共同喜爱的餐厅，且该餐厅索引和最小，意思就是可能有很多共同喜爱的餐厅，但要找索引和最小的。首先把list1中的元素和索引作为键值对存入HashMap中，然后遍历list2，若list2的元素存在于HashMap中，计算两个索引和，定义List集合存储共同喜爱的餐厅，若索引和小于当前索引和，则清空集合，把具有更小索引和的共同餐厅存入集合，若索引和等于当前索引和，则直接存入，最后把集合里的元素复制到数组中并返回。代码如下：

class Solution {
    public String[] findRestaurant(String[] list1, String[] list2) {
        Map<String,Integer> hm=new HashMap<String,Integer>();
        
        //把list1中的元素和索引作为键值对存入HashMap集合
        for(int i=0;i<list1.length;i++){
            hm.put(list1[i],i);
        }
        
        //定义List集合存储共同喜爱的餐厅
        List<String> al=new ArrayList<String>();
        int min=list1.length+list2.length;
        for(int i=0;i<list2.length;i++){
            
            //遍历list2，判断list2元素是否存在于HashMap中
            if(hm.containsKey(list2[i])){
                
                //若存在且索引和比当前索引和小
                if(i+hm.get(list2[i])<min){
                    al.clear();    //清空集合
                    al.add(list2[i]);    //将具有更小索引和的共同餐厅加入
                    min=i+hm.get(list2[i]);
                }else if(i+hm.get(list2[i])==min){
                    al.add(list2[i]);    //若与当前索引和相等则直接存入
                }
            }
        }

        //最后把集合中的元素复制到数组中并返回
        String[] res=new String[al.size()];
        for(int i=0;i<al.size();i++){
            res[i]=al.get(i);
        }
        return res;

        /*若想使代码变得简短些，最后这一步可直接写成
            return al.toArray(new String[al.size()]);*/
    }
}