72 lines
1.6 KiB
Python
72 lines
1.6 KiB
Python
from string import *
|
|
from re import *
|
|
from datetime import *
|
|
from collections import *
|
|
from heapq import *
|
|
from bisect import *
|
|
from copy import *
|
|
from math import *
|
|
from random import *
|
|
from statistics import *
|
|
from itertools import *
|
|
from functools import *
|
|
from operator import *
|
|
from io import *
|
|
from sys import *
|
|
from json import *
|
|
from builtins import *
|
|
import string
|
|
import re
|
|
import datetime
|
|
import collections
|
|
import heapq
|
|
import bisect
|
|
import copy
|
|
import math
|
|
import random
|
|
import statistics
|
|
import itertools
|
|
import functools
|
|
import operator
|
|
import io
|
|
import sys
|
|
import json
|
|
from typing import *
|
|
|
|
# @leet start
|
|
class WordDistance:
|
|
|
|
wordsDistance = []
|
|
def __init__(self, wordsDict: List[str]):
|
|
self.wordsDistance = wordsDict
|
|
self.word_indices = {}
|
|
for i, word in enumerate(wordsDict):
|
|
self.word_indices.setdefault(word, []).append(i)
|
|
|
|
def shortest(self, word1: str, word2: str) -> int:
|
|
pos1 = self.word_indices[word1]
|
|
pos2 = self.word_indices[word2]
|
|
i, j = 0, 0
|
|
min_distance = float("inf")
|
|
|
|
while i < len(pos1) and j < len(pos2):
|
|
index1 = pos1[i]
|
|
index2 = pos2[j]
|
|
min_distance = min(min_distance, abs(index1 - index2))
|
|
|
|
# Move the pointer pointing to the smaller index
|
|
if index1 < index2:
|
|
i += 1
|
|
else:
|
|
j += 1
|
|
|
|
return min_distance
|
|
|
|
|
|
|
|
|
|
# Your WordDistance object will be instantiated and called as such:
|
|
# obj = WordDistance(wordsDict)
|
|
# param_1 = obj.shortest(word1,word2)
|
|
# @leet end
|