1、题目
假设 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)。
2、思路
1、首先提示中说数组的元素唯一,记录较小索引联想到HashMap(和Python字典相同)
2、先按照 key元素:value索引的形式把list1入Map
3、然后遍历list2,查找Map: 如果list2中元素和list1中有相同的,就和当前最小值做对比,当前更小就清空List再add进去。这里清空历史最小值很关键。
4、如果和当前最小值相同就追加
5、最后有一个细节就是List转数组。
如果只是list.toArray(),这样的得到的Array数组是Object类型的。点击toArray看源码可以发现给了实例:也就是说如果转String数组,就要再new一个list.size()大小的String数组传到toArray中。就像下面代码所示:resultList.toArray(new String[resultList.size()]);
3、代码
class Solution {
public String[] findRestaurant(String[] list1, String[] list2) {
Map<String, Integer> tempMap = new HashMap<>();
List<String> resultList = new ArrayList<>();
for (int i = 0; i < list1.length; i++) { //list1入map
tempMap.put(list1[i], i);
}
int minIndex = list1.length + list2.length ;
for (int i = 0; i < list2.length; i++) {
// 如果list2中元素和list1中有相同的,就和当前最小值做对比,当前更小就清空List再add进去
if (tempMap.get(list2[i]) != null && tempMap.get(list2[i])+i < minIndex) {
minIndex = tempMap.get(list2[i])+i;
resultList.clear();
resultList.add(list2[i]);
// 相同就追加
}else if (tempMap.get(list2[i]) != null && tempMap.get(list2[i])+i == minIndex){
resultList.add(list2[i]);
}
}
return resultList.toArray(new String[resultList.size()]);
}
}