题目
假设 Andy 和 Doris 想在晚餐时选择一家餐厅,并且他们都有一个表示最喜爱餐厅的列表,每个餐厅的名字用字符串表示。你需要帮助他们用最少的索引和找出他们共同喜爱的餐厅。 如果答案不止一个,则输出所有答案并且不考虑顺序。 你可以假设答案总是存在。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/minimum-index-sum-of-two-lists
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
- 用哈希表存放第一个列表的元素,再对第二个列表进行一次遍历,记录最小索引和与餐厅名
- 使用 a n s ans ans 存放结果,最终转为字符串数组即可.
代码
class Solution {
public String[] findRestaurant(String[] list1, String[] list2) {
Map<String,Integer> map = new HashMap<String,Integer>(list1.length);
for(int i = 0;i < list1.length; i++){
map.put(list1[i],i);
}
int min = 2000;
int temp;
List<String> ans = new ArrayList<>();
for(int i = 0;i < list2.length; i++){
if(map.containsKey(list2[i])){
temp = map.get(list2[i]);
if(min > temp+i){
ans.clear();
ans.add(list2[i]);
min = temp+i;
}else if(min == i+temp){
ans.add(list2[i]);
}
}
}
return ans.toArray(new String[ans.size()]);
}
}