假设 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)
方法一:哈希表
思路与算法
使用一个哈希表记录 list1中每个餐厅对应的索引下标,然后遍历 list2,如果 list2中的餐厅存在于哈希表中,那么说明该餐厅是两人共同喜爱的,计算它的索引和。如果该索引和比最小索引和小,则清空结果,将该餐厅加入结果中,该索引和作为最小索引和;如果该索引和等于最小索引和,则直接将该餐厅加入结果中。
public class Solution17A {
public String[] findRestaurant(String[] list1, String[] list2) {
//定义一个哈希表
HashMap<String,Integer> index = new HashMap<String,Integer>();
for(int i=0;i<list1.length;i++){
index.put(list1[i],i);
}
//结果集
ArrayList<String> ret = new ArrayList<>();
int indexSum = Integer.MAX_VALUE;
for (int i=0;i<list2.length;i++){
//和哈希表进行对比 如果list2 中有list1
if(index.containsKey(list2[i])){
Integer j = index.get(list2[i]);
//计算索引和 小于最小索引 清空结果集,改索引和作为最小索引和
if(i+j<indexSum){
ret.clear();
ret.add(list2[i]);
indexSum = i+j;
}else if (i+j ==indexSum){// 相等 直接添加进去
ret.add(list2[i]);
}
}
}
return ret.toArray(new String[ret.size()]);
}
}