Leetcode 245 最短单词距离III

给定一个单词列表和两个单词 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 

思路：跟I唯一的区别就是word1和word2有可能是相同的，用p1代表word1所在的位置，p2代表word2所在位置。实时更新p1和p2，如果word1==word2,则把新的位置赋给p2, 把p2赋给p1。

class Solution(object):
    def shortestWordDistance(self, words, word1, word2):
        res = float('inf')
        p1 = len(words)
        p2 = -len(words)
        for i, word in enumerate(words):
            if word == word1:
                p1 = i if word1 != word2 else p2
            if word == word2:
                p2 = i
            res = min(res,abs(p1-p2))
        return res