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

题目:

假设 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()]);

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

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值