【LeetCode】599. 两个列表的最小索引总和

1.题目

假设Andy和Doris想在晚餐时选择一家餐厅，并且他们都有一个表示最喜爱餐厅的列表，每个餐厅的名字用字符串表示。

你需要帮助他们用最少的索引和找出他们共同喜爱的餐厅。 如果答案不止一个，则输出所有答案并且不考虑顺序。 你可以假设总是存在一个答案。
点击跳转原文

2.思路

把list1的元素和索引放入map中；
在lsit2中寻找map中存在的元素；
计算索引之和并进行比较，

3.代码

vector<string> findRestaurant(vector<string>& list1, vector<string>& list2){
	int l1=list1.size(),l2=list2.size();
	map<string,int>mp;
	vector<string>res;
	int minVal=10000; 
	for(int i=0;i<l1;i++){
		mp.insert(pair<string,int>(list1[i],i));//把list1的元素和索引放入map中 
	}
	for(int i=0;i<l2;i++){
		if(mp.find(list2[i])!=mp.end()){//在map中找到list2中的元素，比较索引之和的大小 
			if(mp[list2[i]]+i<minVal){
				minVal=mp[list2[i]]+i;
				res.clear();
				res.push_back(list2[i]);
			}
			else if(mp[list2[i]]+i==minVal){
				res.push_back(list2[i]);
			}
		}
	}
	return res;
}