提示:
1 <= paths.length <= 100
paths[i].length == 2
1 <= cityAi.length, cityBi.length <= 10
cityAi != cityBi
- 所有字符串均由大小写英文字母和空格字符组成。
数据量很小,所以怎么做应该都不会超时。
思路
第一反应是开数组来插入数据但好像很麻烦,也不太会写了。
然后觉得链表好像插入会更方便,但也忘记了。
然后ai给了一个思路是,首先存储起点在一个集合中;然后遍历paths[i][1]如果不在起点集合中,则代表这个站点是终点城市。
代码1(集合set)
class Solution {
public:
string destCity(vector<vector<string>>& paths) {
set<string> startCities; //存储起始城市
for(const auto& path : paths){
startCities.insert(path[0]);
}
for(const auto& path:paths){
if(startCities.find(path[1]) == startCities.end()){
//重点城市不在起始城市集合中,它就是终点站
return path[1];
}
}
return "";
}
};
//find()如果找到会返回指向该城市的迭代器;如果没有找到find会返回集合的尾部迭代器end(),这指向集合最后一个元素的下一个位置,通常表示集合中没有找到这个元素。
代码2(数组)
简单易懂版
class Solution {
public:
string destCity(vector<vector<string>>& paths) {
int n = paths.size();
int i=0,j=0;
string stations[n];
for(i=0;i<n;i++){
stations[i] = paths[i][0]; //保存起点
}
for(i=0;i<n;i++){
for(j=0;j<n;j++){
if(paths[i][1]!=stations[j]) continue;
else break;
}
if(j==n) return paths[i][1];
}
return "";
}
};