假设 Andy 和 Doris 想在晚餐时选择一家餐厅,并且他们都有一个表示最喜爱餐厅的列表,每个餐厅的名字用字符串表示。
你需要帮助他们用最少的索引和找出他们共同喜爱的餐厅。 如果答案不止一个,则输出所有答案并且不考虑顺序。 你可以假设答案总是存在。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/minimum-index-sum-of-two-lists
解答一:暴力循环
class Solution {
public:
vector<string> findRestaurant(vector<string>& list1, vector<string>& list2) {
int m = list1.size();
int n = list2.size();
int record = INT_MAX;
vector<string> ans;
for(int i = 0; i < m; i++){
for(int j = 0; j < n; j++){
if(list1[i] == list2[j]){
if(i + j < record){
record = i + j;
if(!ans.empty()) ans.pop_back();
ans.emplace_back(list1[i]);
}
else if(i + j == record){
ans.emplace_back(list1[i]);
}
}
}
}
return ans;
}
};
解答二:哈希表
用哈希表记录列表1的每个索引 然后在列表二中寻找
class Solution {
public:
vector<string> findRestaurant(vector<string>& list1, vector<string>& list2) {
unordered_map<string, int> cnt;
for(int i = 0; i < list1.size(); i++){
cnt[list1[i]] = i;
}
int record = INT_MAX;
vector<string> ans;
for(int j = 0; j < list2.size(); j++){
if(cnt.count(list2[j])){
int i = cnt[list2[j]];
if(i + j < record){
record = i + j;
ans.clear();
ans.emplace_back(list2[j]);
}
else if(i + j == record){
ans.emplace_back(list2[j]);
}
}
}
return ans;
}
};