LeetCode 599 两个列表的最小索引总和

204 阅读1分钟

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

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

示例 1:

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

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


/**
 * @param {string[]} list1
 * @param {string[]} list2
 * @return {string[]}
 */
 
const findRestaurant = (list1, list2) => {
  let map = new Map()
  for (let i = 0; i < list1.length; i++) {
    for (let j = 0; j < list2.length; j++) {
      let curr = list1[i]
      if (curr === list2[j]) {
        let arr = map.get(i+j)
        if (arr) arr.push(curr)
        else arr = [curr]
        map.set(i+j, arr)
        break
      }
    }
  }
  let [...res] = map.keys()
  return map.get(Math.min.apply(null, res))

对代码进行优化后 时间复杂度降低

/**
 * @param {string[]} list1
 * @param {string[]} list2
 * @return {string[]}
 */
var findRestaurant = function(list1, list2) {
let map = new Map(), arr = [] ,min=Infinity
for (let i = 0;i<list1.length;i++){
    map.set(list1[i],i)
}
for ( let i = 0;i<list2.length;i++){
    let j = map.get(list2[i])
    if(i+j<min){
        min = i+j
        arr = [list2[i]]
    }else if (i+j == min) {
        arr.push(list2[i])
    }
}
return arr
};