leetcode刷题-599两个列表的最小索引总和

435 阅读2分钟

前言

记录下leetcode每日一题

题目描述

假设 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 中的所有字符串都是 唯一 的。

思路

类似查重,可以创建一个表用来存储第一个人的餐厅列表和下标。对第二个的餐厅列表进行遍历,查找当前表中是否存在。这里可以创建一个变量来存储当前两者的最小索引,如果两者共有的餐厅都下标小于最小索引,那么认为这个是最小索引且喜欢的餐厅,如果等于最小索引,那么认为满足条件的餐厅可以有多个

代码

/**
 * @param {string[]} list1
 * @param {string[]} list2
 * @return {string[]}
 */
var findRestaurant = function (list1, list2) {
    const hash = new Map()
    let result = []
    // 将第一个遍历到的数据通过下标存储到hash中
    list1.forEach((item, index) => {
        hash.set(item, index)
    })
    let minIndex = Number.MAX_VALUE // 存储当前最小索引
    list2.forEach((item, index) => {
        if (hash.has(item)) {
            if (index + hash.get(item) < minIndex) {
                // 判断他们的最小索引是否小于当前记录的
                result.length = 0 // 清空之前的数据
                minIndex = index + hash.get(item) // 重新计算当前最小索引
                result.push(item) // 存储公共餐厅
            } else if (index + hash.get(item) == minIndex) {
                result.push(item) // 这里代表多个共同喜爱
            }
        }
    })
    return result
};

最后

每天进步一点点