LeetCode 245. 最短單詞距離 III

1. 題目

給定一個單詞列表和兩個單詞 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
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

2. 解題

類似題目：

LeetCode 243. 最短單詞距離
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

---

我的CSDN博客地址 https://michael.blog.csdn.net/

長按或掃碼關注我的公衆號（Michael阿明），一起加油、一起學習進步！