题目链接:
力扣https://leetcode-cn.com/problems/minimum-index-sum-of-two-lists/
【分析】提示中的这个字符串唯一很重要,这样就可以用哈希表了,具体方法是先把其中一列放进哈希表(key为餐厅名,value为下标),然后遍历另一列,如果当前元素在哈希表中,那么计算下标和。需要注意索引最小的字符串可能有多个,所以答案先用链表存再转字符串数组。
class Solution {
public String[] findRestaurant(String[] list1, String[] list2) {
Map<String, Integer> map = new HashMap();
int n = list1.length, m = list2.length;
int i, j;
for(i = 0; i < n; i++){
map.put(list1[i], i);
}
int min = 20001, t;
List<String> list = new ArrayList();
for(i = 0; i < m; i++){
if(map.containsKey(list2[i])){
t = i + map.get(list2[i]);
if(t < min){
min = t;
list.clear();
list.add(list2[i]);
}else if(t == min){
list.add(list2[i]);
}
}
}
n = list.size();
String[] ret = new String[n];
i = 0;
for(String st: list){
ret[i++] = st;
}
return ret;
}
}