给定一个单词列表和两个单词 word1 和 word2,返回列表中这两个单词之间的最短距离。
word1 和 word2 是有可能相同的,并且它们将分别表示为列表中两个独立的单词。
示例:
假设 words = ["practice", "makes", "perfect", "coding", "makes"].
输入: word1 = “makes”, word2 = “coding”
输出: 1
输入: word1 = "makes", word2 = "makes"
输出: 3
注意:
你可以假设 word1 和 word2 都在列表里。
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/shortest-word-distance-iii 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
类似题目:
LeetCode 244. 最短单词距离 II(哈希map+set二分查找)
程序员面试金典 - 面试题 17.11. 单词距离(multimap平衡二叉搜索树)
class Solution {
public:
int shortestWordDistance(vector<string>& words, string word1, string word2) {
int p1 = -1, p2 = -1, mindis = INT_MAX;
for(int i = 0; i < words.size(); ++i)
{
if(word1 == word2 && words[i] == word1)
{ //相同的情况
if(p1 != -1)
mindis = min(mindis, i-p1);
p1 = i;
continue;
}
//不同的情况
if(words[i] == word1)
p1 = i;
if(words[i] == word2)
p2 = i;
if(p1 != -1 && p2 != -1)
mindis = min(mindis, abs(p1-p2));
}
return mindis;
}
};
20 ms 10.8 MB