0
点赞
收藏
分享

微信扫一扫

算法笔记(599. 两个列表的最小索引总和)

日月同辉9908 2022-04-18 阅读 42
java

题目:

假设 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)。
https://leetcode-cn.com/problems/minimum-index-sum-of-two-lists/

解法一:单map

public String[] findRestaurant(String[] list1, String[] list2) {
        Map<String,Integer> map = new HashMap<>();
        int l1 = list1.length;
        int l2 = list2.length;
        for (int i=0;i<l1;i++){
            map.put(list1[i],i);
        }
        List<String> reList = new ArrayList<>();
        int min = Integer.MAX_VALUE;
        for (int i=0;i<l2;i++){
            if (map.containsKey(list2[i])){
                int j = map.get(list2[i]);
                if (i + j < min) {
                    reList.clear();
                    reList.add(list2[i]);
                    min = i + j;
                } else if (i + j == min) {
                    reList.add(list2[i]);
                }
            }
        }

        return reList.toArray(new String[reList.size()]);

    }

解法二:双map

  // 去重
        Map<String,Integer> disMap = new HashMap<>();
        // 比较
        Map<String,Integer> comMap = new HashMap<>();
        int l1 = list1.length;// 4 4 8
        int l2 = list2.length;
        for (int i=0;i<l1;i++){
            disMap.put(list1[i],1);
            comMap.put(list1[i],i);
        }
        for (int i=0;i<l2;i++){
            disMap.put(list2[i],disMap.getOrDefault(list2[i],0)+1);
            comMap.put(list2[i],comMap.getOrDefault(list2[i],0)+i);
        }

        Iterator<Map.Entry<String, Integer>> disIterator = disMap.entrySet().iterator();
        while (disIterator.hasNext()){
            Map.Entry<String, Integer> entry = disIterator.next();
            if (entry.getValue()==1) disIterator.remove();
        }

        // 通过两次循环 将两个数组中的元素导入map中
        // 现在map中包含了 餐厅名,索引和
        int min = Integer.MAX_VALUE;
        Iterator<Map.Entry<String, Integer>> it = comMap.entrySet().iterator();
        while (it.hasNext()){
            Map.Entry<String, Integer> next = it.next();
            if (disMap.containsKey(next.getKey())){
                min = Math.min(next.getValue(),min);
            }else {
                it.remove();
            }
        }
        // 找到min之后 再次遍历 找到相同大小的 之后放入String[]
        List<String> reList = new ArrayList<>();
        for (String s : comMap.keySet()) {
            if (comMap.get(s)==min){
                reList.add(s);
            }
        }

        return reList.toArray(new String[reList.size()]);

人在状态不好的时候做算法题就容易胡来

举报

相关推荐

0 条评论